BZOJ1397 : Ural 1486 Equal squares

二分答案mid，然后检验是否存在两个相同的mid*mid的正方形

检验方法：

首先对于每个位置，求出它开始长度为mid的横行的hash值

然后对于hash值再求一次竖列的hash值

将第二次求出的hash值排序，如果存在两个相同的hash值则可行

 

#include<cstdio>

#include<algorithm>

#define N 510

typedef unsigned long long ll;

const ll D1=97,D2=131;

int n,m,i,j,l,r,mid,ans,t;char a[N][N];ll pow1[N],pow2[N],h[N][N],tmp,tmp2,hash[N*N];

bool check(int x){

  for(i=1;i<=n;i++){

    for(tmp=0,j=1;j<x;j++)tmp=tmp*D1+a[i][j],h[i][j]=0;

    for(j=x;j<=m;j++)h[i][j]=tmp=tmp*D1-pow1[x]*a[i][j-x]+a[i][j];

  }

  for(t=0,i=x;i<=m;i++){

    for(tmp=0,j=1;j<x;j++)tmp=tmp*D2+h[j][i];

    for(j=x;j<=n;j++)hash[t++]=tmp=tmp*D2-pow2[x]*h[j-x][i]+h[j][i];

  }

  for(std::sort(hash,hash+t),i=1;i<t;i++)if(hash[i-1]==hash[i])return 1;

  return 0;

}

int main(){

  scanf("%d%d",&n,&m);

  for(i=1;i<=n;i++)for(scanf("%s",a[i]+1),j=1;j<=m;j++)a[i][j]-='a'-1;

  l=1,r=n<m?n:m;

  for(pow1[0]=pow2[0]=i=1;i<=r;i++)pow1[i]=pow1[i-1]*D1,pow2[i]=pow2[i-1]*D2;

  while(l<=r)if(check(mid=(l+r)>>1))l=(ans=mid)+1;else r=mid-1;

  return printf("%d",ans),0;

}