蓝桥杯 算法提高 ADV-224 9-1九宫格

算法提高 9-1九宫格
时间限制：1.0s 内存限制：256.0MB
提交此题
问题描述
　　九宫格。输入1-9这9个数字的一种任意排序，构成3*3二维数组。如果每行、每列以及对角线之和都相等，打印1。否则打印0。
样例输出
与上面的样例输入对应的输出。
例：
　输入
4 9 2
3 5 7
8 1 6
　输出
1
数据规模和约定
　　输入1-9这9个数字的一种任意排序。

分析：每行每列每条对角线之和都相等，只需要第一行的和，第一列的和，对角线的和相等即可。不用把三行三列四个对角线都求出来，代码如下：

#include <iostream>
using namespace std;

//判断三个和是否相等
int fun(int *sum)
{
	return sum[0] == sum[1] && sum[1] == sum[2];
}

int main()
{
	int a[3][3];
	int sum[3] = {0};
	
	for(int i = 0; i < 3; i++)
	{
		for(int j = 0; j < 3; j++)
		{
			cin >> a[i][j];
		}
	}
	
	for(int i = 0; i < 3; i++) sum[0] += a[0][i];	//第一列的和
	for(int i = 0; i < 3; i++) sum[1] += a[i][0];	//第一行的和
	for(int i = 0; i < 3; i++) sum[2] += a[i][i];	//对角线的和
	
	cout <<  fun(sum) << endl;
		
	return 0;
}