【蓝桥杯】Python实现蛇行矩阵

【蓝桥杯】蛇行矩阵

时间限制: 1Sec 内存限制: 64MB

题目描述

蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。输入本题有多组数据，每组数据由一个正整数N组成。（N不大于100）输出对于每一组数据，输出一个N行的蛇形矩阵。两组输出之间不要额外的空行。矩阵三角中同一行的数字用一个空格分开。行尾不要多余的空格。

样例输入

5

样例输出

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

PYTHON代码

##蛇行矩阵
size=int(input())
a=[[0 for i in range(size)] for j in range(size)]#生成一个size行size列的元素全为0的矩阵
count=1
for i in range(size):#观察输出的矩阵可得到一定的规律
    for j in range(i,-1,-1):
        a[j][i-j]=count
        count+=1
for i in a:
    s=''
    for j in i:
        if j!=0:
            s=s+str(j)+' '
    print(s.strip())#题目要求最后不带空格，所以用strip()函数去掉两边空格