蓝桥杯试题 算法提高 9-1九宫格 C/C++

试题 算法提高 9-1九宫格

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　九宫格。输入1-9这9个数字的一种任意排序，构成3*3二维数组。如果每行、每列以及对角线之和都相等，打印1。否则打印0。
样例输出
与上面的样例输入对应的输出。
例：

数据规模和约定
　　输入1-9这9个数字的一种任意排序。

思路：按照题目意思分别求出每一行、每一列和两条对角线的和，再将其值进行比较，如果全部都相等输出1，否则输出0。

代码如下：

#include<iostream>
using namespace std;
int main(){
	int a[3][3],b[9],k=0;
	for(int i=0;i<3;i++){
		for(int j=0;j<3;j++){
			cin>>a[i][j];
		}
	}
	b[0]=a[0][0]+a[0][1]+a[0][2];
	b[1]=a[1][0]+a[1][1]+a[1][2];
	b[2]=a[2][0]+a[2][1]+a[2][2];
	b[3]=a[0][0]+a[1][0]+a[2][0];
	b[4]=a[0][1]+a[1][1]+a[2][1];
	b[5]=a[0][2]+a[1][2]+a[2][2];
	b[6]=a[0][0]+a[1][1]+a[2][2];
	b[7]=a[0][2]+a[1][1]+a[2][0];
	for(int i=0;i<7;i++){
		if(b[i]!=b[i+1]){
			k=1;
			break;
		}
	}
	if(k==1){
		cout<<"0";
	} else{
		cout<<"1";
	}
}