ZZULIOJ 1125: 上三角矩阵的判断，Java

1125: 上三角矩阵的判断

题目描述

编写程序，输入一个正整数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

import java.util.Scanner;
public class Main {
    static Scanner cin = new Scanner(System.in);
    static int [][]a = new int[21][21];
    public static void main(String[] args) {
        int n=cin.nextInt();
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                a[i][j]=cin.nextInt();

            int flag=0;
            for(int i=1;i<=n;i++)
                for(int j=1;j<i;j++)
                    if(a[i][j]!=0)
                    {
                        flag=1;
                        break;
                    }
            if(flag==1) System.out.println("NO");
            else System.out.println("YES");
    }
}