1125: 上三角矩阵的判断

题目描述

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

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

int IsUpperTriMatrix(int a[][], 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

以下是c++代码：

#include<iostream>
#include<string>
using namespace std;
int a[10][10];
int IsUpperTriMatrix(int n);
int main()
{
	int n;
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			cin >> a[i][j];
		}
	}
	int judge = IsUpperTriMatrix(n);
	if (judge)
	{
		cout << "YES" << endl;
	}
	else
	{
		cout << "NO" << endl;
	}
	return 0;
}
int IsUpperTriMatrix(int n)
{
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j !=i; j++)
		{
				if (a[i][j] != 0)
				{
					return 0;
				}
		}
	}
	return 1;
}