python　编写题解 1097: 蛇行矩阵

解题思路:用一个函数控制，避免索引越界
参考代码:

def f(a,n):
   li=[a]
   for i in range(num[-1],n+1):
       li.append(li[-1]+i)
   return li
n=int(input())
num=[x for x in range(n,1,-1)]
list=[]
list.append(f(1,n))
num.pop()
li=list[0]
for i in li[:-2]:
   list.append(f(i+1,n))
   num.pop()
list.append([li[-2]+1])

for i in list:
   for j in i:
       print(j,end=" ")
   print()