POJ 2185 Milking Grid（最小重复字符矩阵KMP）

Milking Grid

#include<iostream>
using namespace std;
const int N=2010;

char s[10005][80];
int n,m;
int ne[10005];
bool samr(int i,int j)
{
    for(int k=1;k<=m;k++)
        if(s[i][k]!=s[j][k]) return 0;
    return 1;
}
bool samc(int i,int j)
{
    for(int k=1;k<=n;k++)
        if(s[k][i]!=s[k][j]) return 0;
    return 1;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>s[i]+1;
    
    for(int i=2,j=0;i<=n;i++)
    {
        while(j&&!samr(i,j+1)) j=ne[j];
        if(samr(i,j+1)) j++;
        ne[i]=j;
    }
    int ans=n-ne[n];
    for(int i=2,j=0;i<=m;i++)
    {
        while(j&&!samc(i,j+1)) j=ne[j];
        if(samc(i,j+1)) j++;
        ne[i]=j;
    }
    ans*=m-ne[m];
    cout<<ans<<'\n';
    return 0;
}

F. 地图压缩

q组询问，每次询问一个矩形的最小重复字符矩阵

行列hash，为了在KMP加速匹配过程。

$O(n^2+q^n)$

#include<iostream>
using namespace std;
typedef unsigned long long ull;
const int N=2010;

const ull P=13331;
int n,q;
char s[N][N];
ull hr[N][N],hc[N][N],p[N];
int ne[N];
bool samc(int i,int j,int l,int r)
{
    ull x=hc[i][r]-hc[i][l-1]*p[r-l+1];
    ull y=hc[j][r]-hc[j][l-1]*p[r-l+1];
    return (x==y);
}
bool samr(int i,int j,int l,int r)
{
    ull x=hr[i][r]-hr[i][l-1]*p[r-l+1];
    ull y=hr[j][r]-hr[j][l-1]*p[r-l+1];
    return (x==y);
}
int main()
{
    cin>>n>>q;
    for(int i=1;i<=n;i++) cin>>s[i]+1;
    p[0]=1;
    for(int i=1;i<=n;i++) p[i]=p[i-1]*P;
    
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++) hr[i][j]=hr[i][j-1]*P+s[i][j];
    for(int j=1;j<=n;j++)
        for(int i=1;i<=n;i++) hc[j][i]=hc[j][i-1]*P+s[i][j];
    
    while(q--)
    {
        int a,b,c,d;
        cin>>a>>b>>c>>d;
        
        for(int i=2,j=0;i<=d-b+1;i++)
        {
            while(j&&!samc(i+b-1,j+b,a,c)) j=ne[j];
            if(samc(i+b-1,j+b,a,c)) j++;
            ne[i]=j;
        }
        int ans=d-b+1-ne[d-b+1];
        for(int i=2,j=0;i<=c-a+1;i++)
        {
            while(j&&!samr(i+a-1,j+a,b,d)) j=ne[j];
            if(samr(i+a-1,j+a,b,d)) j++;
            ne[i]=j;
        }
        ans*=c-a+1-ne[c-a+1];
        cout<<ans<<'\n';
    }
    
    return 0;
}