HDU 4029 Distinct Sub-matrix(后缀数组)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4029

题意：给定一个n*m的字母矩阵。求有多少个不同的子矩阵？

思路：枚举列数w，hash[i][j]保存第i行从第j列开始宽度为w的串的哈希值，然后以列为串，也就是有m-w+1个串建立自动机。求不同子串就是宽度为w时的不同的矩阵个数。

int r[N],sa[N],wa[N],wb[N],wd[N],rank[N],h[N];



int cmp(int *r,int a,int b,int len)

{

    return r[a]==r[b]&&r[a+len]==r[b+len];

}



void da(int *r,int *sa,int n,int m)

{

    int i,j,p,*x=wa,*y=wb,*t;

    FOR0(i,m) wd[i]=0;

    FOR0(i,n) wd[x[i]=r[i]]++;

    FOR1(i,m-1) wd[i]+=wd[i-1];

    FORL0(i,n-1) sa[--wd[x[i]]]=i;

    for(j=1,p=1;p<n;j<<=1,m=p)

    {

        p=0;

        FOR(i,n-j,n-1) y[p++]=i;

        FOR0(i,n) if(sa[i]>=j) y[p++]=sa[i]-j;

        FOR0(i,m) wd[i]=0;

        FOR0(i,n) wd[x[i]]++;

        FOR1(i,m-1) wd[i]+=wd[i-1];

        FORL0(i,n-1) sa[--wd[x[y[i]]]]=y[i];

        t=x;x=y;y=t;p=1;x[sa[0]]=0;

        FOR1(i,n-1) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;

    }

}







void calHeight(int *r,int *sa,int n)

{

    int i,j,k=0;

    FOR1(i,n) rank[sa[i]]=i;

    FOR0(i,n)

    {

        if(k) k--;

        j=sa[rank[i]-1];

        while(i+k<n&&j+k<n&&r[i+k]==r[j+k]) k++;

        h[rank[i]]=k;

    }

}





int n,m,b[N];

char s[N],s1[140][140];

u64 hash[140][140];

map<u64,int> mp;

int Max;



void init(int w)

{

    mp.clear(); Max=0;

    int i,j,k;

    FOR0(i,n) FOR0(j,m-w+1)

    {

        hash[i][j]=0;

        for(k=j;k<j+w;k++)

        {

            hash[i][j]=hash[i][j]*107+s1[i][k];

        }

        if(mp.find(hash[i][j])==mp.end())

        {

            mp[hash[i][j]]=++Max;

        }

    }

}



int M;



u64 getCnt(int colCnt)

{

    u64 ans=(i64)M*(M+1)/2;

    int i;

    FOR1(i,M) ans-=h[i];

    FOR1(i,colCnt)

    {

        M-=n;

        ans-=(i64)(n+1)*M;

        M--;

    }

    return ans;

}



u64 cal(int w)

{

    init(w);

    int L=0,i,j;

    FOR0(j,m-w+1)

    {

        FOR0(i,n) r[L++]=mp[hash[i][j]];

        if(j<m-w) r[L++]=++Max;

    }

    r[L]=0;

    da(r,sa,L+1,Max+1);

    calHeight(r,sa,L);

    M=L;

    u64 temp=getCnt(m-w+1);

    return temp;

}





int main()

{

    int C,num=0;

    RD(C);

    while(C--)

    {

        RD(n,m);

        int i;

        u64 ans=0;

        FOR0(i,n) RD(s1[i]);

        FOR1(i,m) ans+=cal(i);

        printf("Case #%d: %I64u\n",++num,ans);

    }

    return 0;

}