上三角矩阵判定

题目

KiKi想知道一个n阶方矩是否为上三角矩阵，请帮他编程判定。上三角矩阵即主对角线以下的元素都为0的矩阵，主对角线为从矩阵的左上角至右下角的连线。

输入描述:

第一行包含一个整数n，表示一个方阵包含n行n列，用空格分隔。 (1≤n≤10)

从2到n+1行，每行输入n个整数（范围-231~231-1），用空格分隔，共输入n*n个数。

输出描述：

一行，如果输入方阵是上三角矩阵输出"YES"并换行，否则输出"NO"并换行

import java.util.*;

public class Main{
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[][] a = new int[n][n];

        boolean flag = true;
        for (int i=0; i<n; i++){
            for (int j=0; j<n; j++){
                a[i][j] = sc.nextInt();
                if (i > 0 && j < i && a[i][j] != 0)
                    flag = false;
            }
        }

        if (flag)
            System.out.println("YES");
        else
            System.out.println("NO");
    }
}