【C语言】判断是否为上三角矩阵。

【问题描述】判断是否为上三角矩阵。
输入一个正整数n(1 (上三角矩阵,即主对角线以下的元素都为0，主对角线为从矩阵的左上角至右下角的连线)｡
【输入形式】
从键盘输入一个正整数n和矩阵阵列。
【输入输出样例1】
Input n:3
Input array:（输出后换行）
1 2 3
0 4 5
0 0 6
YES
【输入输出样例2】
Input n:3
Input array:（输出后换行）
1 2 3
4 5 6
7 8 9
NO
【样例说明】
输出提示符后冒号为英文字符，后面没有空格。
输出结束不换行。

#include
#define N 10
int main()
{
     
	int i,j,n,count=0;
	int a[N][N];
	printf("Input n:");
	scanf_s("%d",&n);
	printf("Input array:\n");
	for (i = 0; i < n; i++)
	{
     
		for (j = 0; j < n; j++)
			scanf_s("%d",&a[i][j]);
	}
	for (i = 0; i < n; i++)
	{
     
		for (j = 0; j < i; j++)
		{
     
			if (a[i][j] == 0)
				count++;
		}
	}
	if (count == n * (n - 1) / 2)
		printf("YES");
	else
		printf("NO");
	return 0;
}