2536 回文矩阵
给你n2n2个数,问是否能构造成为一个n×nn×n的**回文矩阵**(n≤20n≤20)。
回文矩阵指如果将行的顺序反转,矩阵不变;如果将列的顺序反转,矩阵也不变。
输入
第一行一个整数n,1<=n<=20; 第二行n*n个整数,每个数在[1,1000]内。输出
如果可以,输出“YES”; 否则,输出"NO"。输入样例
3 1 1 1 1 1 3 3 3 3输出样例
YES
思路:
考虑%4==0,和%4==1的情况就行了。
PS:注意奇数情况下回文串的判断不要用num等1,要为奇数。
代码实现:
#include<iostream>
#include<cstring>
#include<cmath>
#include<queue>
#include<cstdio>
#include<algorithm>
#define LL long long
#define INF 0x3f3f3f3f
using namespace std;
const int N=2e5+100;
const int M=4e4+100;
int arr[N],num[N];
int main() {
int n,m;
string ans;
while(cin>>n) {
memset(num,0,sizeof(num));
for(int i=1; i<=n*n; i++) {
cin>>arr[i];
num[arr[i]]++;
}
int tmp=0,cnt=0;
m=unique(arr+1,arr+1+n*n)-arr-1;
for(int i=1; i<=m; i++) {
//num==3的情况不会和正解同时出现
//所以无需%4==1
//PS:这里奇数情况判断是否对称不能用num==1
if(num[arr[i]]&1)tmp++;
if(num[arr[i]]%4==0)cnt++;
}
if(n&1) {
if(tmp==1&&cnt==m-1)ans="YES";
else ans="NO";
} else {
if(tmp==0&&cnt==m)ans="YES";
else ans="NO";
}
cout<<ans<<endl;
}
return 0;
}
THE END;