蛇形填充数组

描述

用数字1,2,3,4,...,n*n这n2个数蛇形填充规模为n*n的方阵。

蛇形填充方法为：

对于每一条左下-右上的斜线，从左上到右下依次编号1,2,...,2n-1；按编号从小到大的顺序，将数字从小到大填入各条斜线，其中编号为奇数的从左下向右上填写，编号为偶数的从右上到左下填写。

比如n=4时，方阵填充为如下形式：

1  2  6  7
3  5  8  13
4  9  12 14
10 11 15 16

输入

输入一个不大于10的正整数n，表示方阵的行数。

输出

输出该方阵，相邻两个元素之间用单个空格间隔。

样例输入

4

样例输出

1 2 6 7
3 5 8 13
4 9 12 14
10 11 15 16

#endcoding : UTF-8
"""
@author = 寻找任大侠
@email = renjx@stu.pku.edu.cn
@create_time = 2021/11/27 18:26
[[1, 2, 4, 7],
[3, 5, 8, 11],
[6, 9, 12, 14],
[10, 13, 15, 16]]

[[1, 2, 6, 7],
[3, 5, 8, 13],
[4, 9, 12, 14],
[10, 11, 15, 16]]

"""
n = int(input())
A = [[0 for i in range(n)] for j in range(n)]
temp = 1
for i in range(2*n-1):
    for j in range(i+1):
        if j<n and i-j<n:
            if i%2 == 0:
                A[i-j][j] = temp
            else:
                A[j][i - j] = temp
            temp+=1
for i in A:
    print(" ".join(list(map(str,i))))