1125: 上三角矩阵的判断 Python

1125: 上三角矩阵的判断
时间限制: 1 Sec 内存限制: 128 MB
提交: 9462 解决: 5920
[状态] [讨论版] [提交] [命题人:admin]
题目描述
编写程序，输入一个正整数n（1<=n<=10）和n阶方阵a中的元素，如果a是上三角矩阵，输出“YES”，否则，输出“NO”。
上三角矩阵即主对角线以下（不包括主对角线）的元素都为0的矩阵，主对角线为从矩阵的左上角至右下角的连线。
要求定义函数IsUpperTriMatrix()判断矩阵a是否是上三角矩阵，如果是返回1，否则返回0。函数原型如下：

int IsUpperTriMatrix(int a[][N], int n);
输入
输入一个正整数n（1<=n<=10）和n阶方阵a中的元素，均为整数。
输出
如果a是上三角矩阵，输出“YES”，否则，输出“NO”
样例输入 Copy
4
1 2 3 4
0 2 2 2
0 0 3 4
0 0 0 4
样例输出 Copy
YES

# 上三角矩阵的判断

n = int(input())
flag = True
for i in range(n):
    a = list(map(int, input().split()))
    for j in range(n):
        # 判断下三角部分是否有元素不为0
        if i > j and a[j] != 0:
            flag = False
            break
if flag:
    print("YES")
else:
    print("NO")