HDU 4436 后缀数组

点击打开链接

题意：给n个串，求所有的子串去重后相加对2012取余的值

思路：思路借鉴这篇博客，其中有一个求一段串可以形成的和有点类似与字符串hash的想法，用的很巧，要用后缀数组的话不同两个串的连接需要用一个其他的数字，然后正常计算sa和height，然后当遇到的第i个是加的那个数字或者是0都不计算，因为题目说前导不能为0，然后i+height[Rank[i]]要小于i所在的串的结束位置，然后计算的式子就是要加的结果，画一画样例就好理解了

#include <stdio.h>#include <stdlib.h>#include <string.h>#include <iostream>#include <algorithm>using namespace std;typedef long long ll;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const ll INF=0x3f3f3f3f3f3f3f3fll;const int MAXN=200010;const int mod=2012;int sa[MAXN];int t1[MAXN],t2[MAXN],c[MAXN];int Rank[MAXN],height[MAXN];void construct_sa(int s[],int n,int m){    int i,j,p,*x=t1,*y=t2;    for(i=0;i<m;i++)c[i]=0;    for(i=0;i<n;i++)c[x[i]=s[i]]++;    for(i=1;i<m;i++)c[i]+=c[i-1];    for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i;    for(j=1;j<=n;j<<=1){        p=0;        for(i=n-j;i<n;i++)y[p++]=i;        for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;        for(i=0;i<m;i++)c[i]=0;        for(i=0;i<n;i++)c[x[y[i]]]++;        for(i=1;i<m;i++)c[i]+=c[i-1];        for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];        swap(x,y);        p=1;x[sa[0]]=0;        for(i=1;i<n;i++)            x[sa[i]]=y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+j]==y[sa[i]+j]?p-1:p++;        if(p>=n)break;        m=p;    }}void construct_lcp(int s[],int n){    int k=0;    for(int i=0;i<=n;i++) Rank[sa[i]]=i;    for(int i=0;i<n;i++){        if(k)k--;        int j=sa[Rank[i]-1];        while(s[i+k]==s[j+k])k++;        height[Rank[i]]=k;    }}int s[MAXN],id[MAXN],sum[MAXN],T[MAXN],V[MAXN],base[MAXN],arrive[MAXN];char str[MAXN];int calc_sum(int l,int r){    int ans=T[r+1]-T[l];    ans%=mod;ans-=V[l]*base[r-l+1];    ans=(ans+mod)%mod;    if(ans<0) ans+=mod;    return ans;}int main(){    base[1]=10;for(int i=2;i<MAXN;i++) base[i]=(base[i-1]+1)*10%mod;    int n;    while(scanf("%d",&n)!=-1){        int k=0,val=0;        for(int i=1;i<=n;i++){            scanf("%s",str);            int len=strlen(str);            for(int j=0;j<len;j++){                s[k]=str[j]-'0'+1;                val=(val*10+s[k]-1)%mod;                V[k+1]=val;                T[k+1]=(T[k]+val)%mod;                id[k]=i;k++;            }            s[k]=11;            val=(val*10+10)%mod;            V[k+1]=val;            T[k+1]=(T[k]+val)%mod;            id[k]=i;k++;arrive[i]=k-2;        }        s[k]=0;        construct_sa(s,k+1,128);        construct_lcp(s,k);        int ans=0;        for(int i=0;i<k;i++){            if(s[i]%10!=1){                if(i+height[Rank[i]]<=arrive[id[i]]){//小于这个界限才会有不重复的串                    ans+=calc_sum(i,arrive[id[i]])-calc_sum(i,i+height[Rank[i]]-1);                    ans=(ans+mod)%mod;                    if(ans<0) ans+=mod;                }            }        }        printf("%d\n",ans);    }    return 0;}