HJ35 蛇形矩阵

题目描述:
蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。

例如,当输入5时,应该输出的三角形为:

1 3 6 10 15

2 5 9 14

4 8 13

7 12

11

输入描述:
输入正整数N(N不大于100)

输出描述:
输出一个N行的蛇形矩阵。

示例1
输入:
4
复制
输出:
1 3 6 10
2 5 9
4 8
7
题解:
通用的公式: (i+j-2)*(i+j-1)/2 + j

代码:

while True:
    try:
        num = int(input())
        for i in range(1,num+1):
            for j in range(1,num-i+2):
                print((i+j-2)*(i+j-1)//2+j, end=" ")
            print(end="\n")
    except:
        break

你可能感兴趣的:(HJ35 蛇形矩阵)