上三角矩阵的判断

标题：上三角矩阵的判断

题目描述
编写程序，输入一个正整数n（1<=n<=10）和n阶方阵a中的元素，如果a是上三角矩阵，输出“YES”，否则，输出“NO”。
上三角矩阵即主对角线以下（不包括主对角线）的元素都为0的矩阵，主对角线为从矩阵的左上角至右下角的连线。

要求定义函数IsUpperTriMatrix()判断矩阵a是否是上三角矩阵，如果是返回1，否则返回0。函数原型如下：

IsUpperTriMatrix(int a, int n);

输入
输入一个正整数n（1<=n<=10）和n阶方阵a中的元素，均为整数。

输出
如果a是上三角矩阵，输出“YES”，否则，输出“NO”

样例输入

4
1 2 3 4
0 2 2 2
0 0 3 4
0 0 0 4 

样例输出

YES

代码如下：

#include<stdio.h>
#include<math.h>
int main()
{
    int i,j,x,n,a[10][10];
    int flag=1;
    scanf("%d",&n);
    for(i=0;i<n;i++)
    {
        for(j=0;j<n;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    for(i=0;i<n;i++)
    {
        for(j=0;j<n;j++)
        {
            if(i>j)
            {
                if(a[i][j!=0])
                {
                    flag=0;
                    break;
                }
            }
        }
    }
    if(flag==1)
    {
        printf("YES");
    }
    if(flag==0)
    {
        printf("NO");
    }
    return 0;
}