C++题目：验证幻方

验证幻方

题目描述

幻方是一类有趣的数阵，它是由若干个整数组成的正方形。对于一个幻方，它每一行、每一列和两条对角线上的数相加之和均为同一个值。

给定一个正方形的数阵，请编程判断它是否是一个幻方。

输入格式

输入由$n+1$行组成：

第一行为一个正整数$n$。

接下来的$n$行为一个$n\times n$的正方形数阵，数据保证此数阵由$1\sim n^2$范围的整数各一个。

输出格式

若此数阵是一个幻方，输出"Yes";否则输出"No"

样例 #1

样例输入 #1

3
8 1 6
3 5 7
4 9 2

样例输出 #1

Yes

提示

$1\le n\le 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;
}