Description
Orez很喜欢搜集一些神秘的数据,并经常把它们排成一个矩阵进行研究。最近,Orez又得到了一些数据,并已经把它们排成了一个n行m列的矩阵。通过观察,Orez发现这些数据蕴涵了一个奇特的数,就是矩阵中上下对称且左右对称的正方形子矩阵的个数。 Orez自然很想知道这个数是多少,可是矩阵太大,无法去数。只能请你编个程序来计算出这个数。
Input
文件的第一行为两个整数n和m。接下来n行每行包含m个正整数,表示Orez得到的矩阵。
Output
文件中仅包含一个整数answer,表示矩阵中有answer个上下左右对称的正方形子矩阵。
Sample Input
5 5
4 2 4 4 4
3 1 4 4 3
3 5 3 3 3
3 1 5 3 3
4 2 1 2 4
Sample Output
27
数据范围
对于30%的数据 n,m≤100
对于100%的数据 n,m≤1000 ,矩阵中的数的大小≤109
解法:
正解是manacher,但是是二维的,所以直接用了邪教hash+二分
、
首先为了避免边长奇偶性带来的WT要把矩阵扩大二倍 然后样例就变成了这样:
00000000000
04020404040
00000000000
03010404030
00000000000
03050303030
00000000000
03010503030
00000000000
04020102040
00000000000
把这个矩阵从四个角各哈希一遍 对于每个点二分答案 验证时将四个哈希值全都取出来对比即可
然后只枚举i+j为偶数的点 得到的边长除以2就是以这个点为中心的正方形回文子矩阵数量
然后,本题卡了unsigned long long 改成unsigned int 就可以完美卡过了。。。。
//BZOJ 1414
#include <bits/stdc++.h>
using namespace std;
const int maxn = 2010;
#define BASE1 100000007
#define BASE2 100000009
typedef unsigned int lint;
lint sum[4][maxn][maxn];
int n, m;
lint ans, power1[maxn], power2[maxn];
bool check(int x, int y, int len)
{
lint hash1, temp;
hash1 = sum[0][x+len-1][y+len-1] - sum[0][x-len][y+len-1]*power1[len+len-1] - sum[0][x+len-1][y-len]*power2[len+len-1]
+sum[0][x-len][y-len]*power1[len+len-1]*power2[len+len-1];
temp = sum[1][x+len-1][y-len+1] - sum[1][x-len][y-len+1]*power1[len+len-1] - sum[1][x+len-1][y+len]*power2[len+len-1]
+sum[1][x-len][y+len]*power1[len+len-1]*power2[len+len-1];
if(temp != hash1) return false;
temp = sum[2][x-len+1][y+len-1] - sum[2][x+len][y+len-1]*power1[len+len-1] - sum[2][x-len+1][y-len]*power2[len+len-1]
+sum[2][x+len][y-len]*power1[len+len-1]*power2[len+len-1];
if(temp!=hash1) return false;
temp = sum[3][x-len+1][y-len+1] - sum[3][x+len][y-len+1]*power1[len+len-1] - sum[3][x-len+1][y+len]*power2[len+len-1]
+sum[3][x+len][y+len]*power1[len+len-1]*power2[len+len-1];
if(temp != hash1) return false;
return true;
}
lint solve(int x, int y)
{
int ret=1;
int l=1, r = min(min(x,n-x+1),min(y,m-y+1));
while(l<=r){
int mid = (l+r)/2;
if(check(x,y,mid)) ret=mid, l=mid+1;
else r=mid-1;
}
return ret;
}
int main()
{
scanf("%d%d", &n,&m);
n=n*2+1, m=m*2+1;
for(int i=2; i<=n; i+=2){
for(int j=2; j<=m; j+=2){
int x;
scanf("%d", &x);
for(int k=0; k<4; k++){
sum[k][i][j]=x;
}
}
}
power1[0]=power2[0]=1;
for(int i=1; i<maxn; i++){
power1[i]=power1[i-1]*BASE1;
power2[i]=power2[i-1]*BASE2;
}
for(int i=1; i<=n; i++){
for(int j=1; j<=m; j++){
sum[0][i][j]+=sum[0][i-1][j]*BASE1;
}
}
for(int i=1; i<=n; i++){
for(int j=1; j<=m; j++){
sum[0][i][j]+=sum[0][i][j-1]*BASE2;
}
}
for(int i=1; i<=n; i++){
for(int j=m; j>=1; j--){
sum[1][i][j] += sum[1][i-1][j]*BASE1;
}
}
for(int i=1; i<=n; i++){
for(int j=m; j>=1; j--){
sum[1][i][j] += sum[1][i][j+1]*BASE2;
}
}
for(int i=n; i>=1; i--){
for(int j=1; j<=m; j++){
sum[2][i][j] += sum[2][i+1][j]*BASE1;
}
}
for(int i=n; i>=1; i--){
for(int j=1; j<=m; j++){
sum[2][i][j] += sum[2][i][j-1]*BASE2;
}
}
for(int i=n; i>=1; i--){
for(int j=m; j>=1; j--){
sum[3][i][j] += sum[3][i+1][j]*BASE1;
}
}
for(int i=n; i>=1; i--){
for(int j=m; j>=1; j--){
sum[3][i][j] += sum[3][i][j+1]*BASE2;
}
}
for(int i=1; i<=n; i++){
for(int j=1; j<=m; j++){
if((i+j)%2==0){
ans += solve(i, j) / 2;
}
}
}
cout<<ans<<endl;
return 0;
}