【牛客网华为机试】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

解题思路

（1）总结第一行数字的规律为，按照下标从0开始

(x+2)*(x+1)//2

（2）第二行为第一行剔除第一个数字之后各元素减1

（3）输出每一行

代码

while True:
    try:
        num = int(input())
        for i in range(num):
            if i == 0:
                res = [(x+2)*(x+1)//2 for x in