1414: [ZJOI2009]对称的正方形 Hash+二分

来源:互联网 发布:windows相对路径写法 编辑:程序博客网 时间:2024/05/20 04:26

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 100000009typedef 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;}
0 0
原创粉丝点击