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

分析

首先归结出判断上三角的函数的条件，定义为一个函数，以函数阶数和矩阵的列表作为参数。这里注意，列表作为参数的定义方法：def shangsanjiao(m,lst=[])
for循环，实现按照输入要求的矩阵个数

题目

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

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

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

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

解法

def shangsanjiao(m,lst=[]):#函数阶数和矩阵的列表作为参数
    for i in range(1,m):
        for j in range(i):
            if lst[i*m+j]!=0:#如果不是0，直接返回False
                return False
    return True
fir=int(input())#接受矩阵个数
for i in range(fir):
    m=int(input())#接受矩阵阶数
    lst=[]
    for j in range(m):
        lst.extend(list(map(int,input().split())))#extend实现连接输入的元素，存到一个矩阵里面。
    if shangsanjiao(m,lst)==True:
        print("YES")
    else:
        print("NO")