题目:
上三角矩阵指主对角线以下的元素都为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
注意:
上三角矩阵题目解释的很清楚,我误以为是只有对角线下全为0,其余地方不能为0,审题不清,导致后面改了好久
思路:
看到这道题时,就想用一个大列表,里面有n个小列表,小列表里装着m行
[[[1, 2, 3], [0, 4, 5], [0, 0, 6]], [[1, 0], [-8, 2]]]
#如输入所示,有2个待检测方阵
#第一个[[1, 2, 3], [0, 4, 5], [0, 0, 6]]
#第二个[[1, 0], [-8, 2]]
代码:
对多维列表的建立不熟悉可能开始不好弄
map(fun(),list[]) map函数可以让后面的list全用前面的fun()函数,
返回的一个迭代器,故要用list列表化
map函数用法
extend可以向目标列表添加后面列表的全部数
list—extend用法
n=int(input())#待测矩阵数
a=[]
for i in range(n):
a.append([])
n1=int(input())#待测矩阵行数
for j in range(n1):
a[i].append([])
a[i][j].extend(list(map(int,input().split())))
for an in a:
flag=True
for i in range(1,len(an)):
for j in range(0,i):
if an[i][j]!=0:
flag=False
if flag==True:
print('YES')
else:
print('NO')