PTA判断上三角矩阵（C语言）

题意：

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

鸣谢中水七局机电安装分局喻春桥补充数据！

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

代码如下（示例）：

---

代码：

#include <stdio.h>
#define MAXN 10
int Init_trueORfalse(int a[][MAXN],int n){//初始数组并判断
    int cnt=0;//计算符合条件的0的个数
    int k=n*(n-1)/2;//当方阵规模为n*n，三角矩阵中符合条件的0应该有k个；
for(int i=0;i<n;i++){//k用数列求和来求，求出通项公式
for(int j=0;j<n;j++){
    scanf("%d",&a[i][j]);
    if(i>j&&a[i][j]==0) cnt++;//符合条件的0的个数累加
     }
   }
    return cnt==k;//若a[][]是三角矩阵，则返回1，否则返回0；
 }
int main(){
    int T,n;
    scanf("%d",&T);
    int c;
    for(int i=0;i<T;i++){//输入数组并输出结果
    scanf("%d",&n);
    int a[n][MAXN];
    c=Init_trueORfalse(a,n);
    if(c) printf("YES\n");
    else printf("NO\n");  
    }
    return 0;
}