PTA python 第4章-21 判断上三角矩阵

第4章-21 判断上三角矩阵

分数 15

全屏浏览题目

切换布局

作者 C课程组

单位 浙江大学

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

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

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

YES
NO
NO

鸣谢中水七局机电安装分局喻春桥补充数据！

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

 

def panduan(x, list0):
    '''
    :param x: 第x+1行
    :return: 本行是否满足上三角矩阵
    '''
    for i in range(x):
        if list0[i] == '0':
            return False
        else:
            return True


n = int(input())
f = True
for i in range(n):
    m = int(input())
    for j in range(m):
        if panduan(j, input().split()) :
            f = False
    if f:
        print('YES')
    else:
        print('NO')
    f = True