c 语言 上三角矩阵的判断

./*****
题目描述
编写程序，输入一个正整数n（1<=n<=10）和n阶方阵a中的元素，如果a是上三角矩阵，输出“YES”，否则，输出“NO”。
上三角矩阵即主对角线以下（不包括主对角线）的元素都为0的矩阵，主对角线为从矩阵的左上角至右下角的连线。
要求定义函数IsUpperTriMatrix()判断矩阵a是否是上三角矩阵，如果是返回1，否则返回0。函数原型如下：

int IsUpperTriMatrix(int a[][N], int n);
输入
输入一个正整数n（1<=n<=10）和n阶方阵a中的元素，均为整数。
输出
如果a是上三角矩阵，输出“YES”，否则，输出“NO”
样例输入 Copy
4
1 2 3 4
0 2 2 2
0 0 3 4
0 0 0 4
样例输出 Copy
YES
*****/

！！！！主要就是二维数组的使用而已，这道题没有什么难度

#include <stdio.h>
#include <stdlib.h>
#define N 10
int IsUpperTriMatrix(int a[][N], int n);//第 i行的前i个元素为0否
int main()
{
    int a[N][N];
    int n,i,j;
    scanf("%d",&n);
    for ( i = 0;i<n;i++ )
    {
        for (j = 0;j<n;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    if(IsUpperTriMatrix(a,n))
    printf("YES");
    else
    printf("NO");
    return 0;
}
int IsUpperTriMatrix(int a[][N], int n)
{
    int i,j,flag;
    flag = 0;
    for (i = 1;i<n;i++)
    {
        for (j = 0;j<i;j++)
        {
            if (a[i][j] != 0)
            {
                flag = 1;
            }
        }
    }
    if (flag == 1)
    return 0;
    else
    return 1;
}