HAUTOJ1125上三角矩形的判断

#include<stdio.h>
#define N 10
void TArrInput(int x,int y,int a[N][N])
{
    int i,j;
    for(i=0;i<x;i++)
        for(j=0;j<y;j++)
            scanf("%d",&a[i][j]);
}
// void TArrOutput(int x,int y,int a[N][N])
// {
//     int i,j;
//     for(i=0;i<x;i++)
//         {
//             for(j=0;j<y;j++)
//                 printf("%d ",a[i][j]);
//                 printf("\n");
//         }
// }
int IsUpperTriMatrix(int a[N][N], int n)
{
    int i,j,flag=1;
    //忘记给flag赋值，导致return一直是0
    for(i=0;i<n;i++)
        for(j=0;j<i;j++)
            if(a[i][j]!=0)
                flag=0;
    if(flag!=0)
    return 1;
    else
    return 0;
}
int main()
{
    int a[N][N];
    int n;
    scanf("%d",&n);
    getchar();
    TArrInput(n,n,a);
    if(IsUpperTriMatrix(a,n)==0)
        printf("NO\n");
    else
        printf("YES\n");
    //TArrOutput(n,n,a);
    return 0;
}