习题7-3 判断上三角矩阵 取巧

习题7-3 判断上三角矩阵 (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

闲来无事，再水一篇。

#include <stdio.h>
int main()
{
    int n;								//矩阵个数 
    scanf("%d",&n);
    for(int i = 0; i < n; ++i)			//控制矩阵的输入，及结果的输出 
    {
        int m,flag = 1; 				//m为矩阵的行列数 ，flag为控制结果输出 
        scanf("%d",&m);	
        int a[m*m];						//定义输入矩阵 
        for(int j = 0; j < m*m; j++)	//矩阵的输入 
            scanf("%d",&a[j]);
        for(int j = 0; j < m*m; j++)	//对矩阵的判断 
        {
        	int b = j/m;				//b为当前矩阵的行数-1 
        	if(j >= b*m && j <b*(m+1) && a[j] != 0)	// 判断依据为大于bm，小于中位线，中位线为1 + b（m+1）,不为0后对flag赋值，直接退出判断 
        	{
        		flag = 0;
        		break;
			}
		}
        if(flag == 1)					//结果输出 ，1为YES 
            printf("YES");
        else 
            printf("NO");
        if(i != (n-1)) printf("\n");	//换行的控制 
    }
    return 0;
}