浙大 PTA Python练习 4-21 判断上三角矩阵

题干：上三角矩阵指主对角线以下的元素都为0的矩阵；主对角线为从矩阵的左上角至右下角的连线。

本题要求编写程序，判断一个给定的方阵是否上三角矩阵。

输入格式：输入第一行给出一个正整数T，为待测矩阵的个数。接下来给出T个矩阵的信息：每个矩阵信息的第一行给出一个不超过10的正整数n。随后n行，每行给出n个整数，其间以空格分隔。

输出格式：每个矩阵的判断结果占一行。如果输入的矩阵是上三角矩阵，输出“YES”，否则输出“NO”。

输入样例：

2
3
1 2 3
0 4 5
0 0 6
2
1 0
-8 2

输出样例：

YES
NO

解答：

T=int(input())#输入需要判断的矩阵数量，已知数量后下一步用for循环遍历每个数组
for k in range(T):
    n=int(input())
    lst=[]
    for i in range(n):
        lst.append(list(map(int,input().split())))#二维列表的建立
    flag=1#先假定其为上三角矩阵，flag为常用的旗帜变量
    for i in range(n):
        for j in range(0,i):#上三角矩阵的特点是主对角线以下全是0，主对角线以下行标永远大于列标
            if lst[i][j