数学规律 | HJ35 蛇形矩阵

# 最优解(每一行的一个元素为等差数列的累加值)
n = int(input().strip())
for i in range(1,n+1) :
    for j in range(i,n+1) :
        print(int(((j+j**2)/2)-i+1),end = ' ')
    print()

# 我的代码
while True:
    try:
        n = int(input())
        # nums = [i for i in range(1, n+1)]
        res = [[0 for _ in range(n)]for _ in range(n)]
        cnt = 1
        for i in range(n):
            tmp_i, tmp_j = i, 0
            while tmp_i >= 0 and tmp_j < n:
                res[tmp_i][tmp_j] = cnt
                cnt += 1
                tmp_i -= 1
                tmp_j += 1
        for row in res:
            print(' '.join([str(col) for col in row if col!=0]))
    except:
        break

用时：30min