浙大版《Python 程序设计》题目集 第4章-21 判断上三角矩阵 (15分)

题目:

上三角矩阵指主对角线以下的元素都为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')
发布了22 篇原创文章 · 获赞 4 · 访问量 874

猜你喜欢

转载自blog.csdn.net/qq_43745026/article/details/104660917