题解 | 蛇形矩阵
蛇形矩阵
https://www.nowcoder.com/practice/649b210ef44446e3b1cd1be6fa4cab5e
import sys
n=int(input())
a=[[0]*n for _ in range(n)]
k=n
for i in range(n):
for j in range(k):
if i==0 and j==0:
a[i][j]=1
elif i!=0 and j==0:
a[i][j]=a[i-1][j]+i+j
else:
a[i][j]=a[i][j-1]+i+j+1
if j==k-1:
print(a[i][j])
else:
print(a[i][j],end=' ')
k-=1