验证幻方
题目描述
幻方是一类有趣的数阵,它是由若干个整数组成的正方形。对于一个幻方,它每一行、每一列和两条对角线上的数相加之和均为同一个值。
给定一个正方形的数阵,请编程判断它是否是一个幻方。
输入格式
输入由 n + 1 n+1 n+1行组成:
第一行为一个正整数 n n n。
接下来的 n n n行为一个 n × n n\times n n×n的正方形数阵,数据保证此数阵由 1 ∼ n 2 1\sim n^2 1∼n2范围的整数各一个。
输出格式
若此数阵是一个幻方,输出"Yes";否则输出"No"
样例 #1
样例输入 #1
3
8 1 6
3 5 7
4 9 2
样例输出 #1
Yes
提示
1 ≤ n ≤ 1000 1\le n \le 1000 1≤n≤1000。
答案
#include<bits/stdc++.h>
using namespace std;
int main()
{
int a[1001][1001],n,h[1001]={0},l[1001]={0},d[3]={0};
int i,j;
bool f=true;
cin >> n;
for(i=1;i<=n;i++){
for(j=1;j<=n;j++){
cin >> a[i][j];
}
}
for(i=1;i<=n;i++){
for(j=1;j<=n;j++){
h[i] += a[i][j];
}
}
for(i=1;i<=n;i++){
for(int j=1;j<=n;j++){
l[i] += a[j][i];
}
}
for(i=1;i<=n;i++){
j=n-i+1;
d[1] += a[i][i];
d[2] += a[i][j];
}
for(i=1;i<=n;i++){
for(j=i+1;j<=n;j++){
if(h[i] != h[j]) f=false;
if(l[i] != l[j]) f=false;
if(d[1] != d[2]) f=false;
}
}
if(h[1] != l[1] or l[1] != d[1] or d[1] != h[1]) f=false;
if(f==true) cout << "Yes";
else cout << "No";
return 0;
}