【PTA】【C语言】【Python】判断上三角矩阵

上三角矩阵指主对角线以下的元素都为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

 先来段C语言代码（OVO）：

#include<stdio.h>
int main()
{
	int m, n, i, j, count;
	scanf("%d",&m);
	for(i = 0; i < m; i++)
    {
		scanf("%d",&n);
		int a[n][n];
		for(int i = 0; i < n; i++)
			for(int j = 0; j < n; j++)
            {
				scanf("%d",&a[i][j]);
            }
		count = 0;
		for(int i = 1; i < n; i++)
			for(int j = 0; j <= i-1; j++)
            {
				if(i == j) continue;
				if(a[i][j]!=0)
                {
					count = 1;
					break;
				}
			}
		if(count == 0) printf("YES\n");
		else printf("NO\n");
	}
}

 再来个Python的：

 

n = int(input())
for k in range(n):
    count = 1
    s =[]
    m = int(input())
    for i in range(m):
        s.append(list(map(int,input().split())))

    for i in range(m):
        for j in range(0,i):
            if s[i][j] != 0:
                count = 0
                break
    if count == 1:
        print("YES")
    else:
        print("NO")