HJ35 蛇形矩阵

思路:

数学规律,看了别的uu的思路写的
step1:输入一个正整数;
step2:两个for循环,对i、j进行遍历,如果到最右端的时候,换行;如果不到最右边的数,以空格隔开

代码如下:

n = int(input())
for i in range(1 , n + 1):
    for j in range(1 , n - i + 2):
        if j == n - i + 1:
            print((i + j -2) * (i + j - 1)//2 + j)
        else:
            print((i + j -2) * (i + j - 1)//2 + j , end = ' ')