hdu 4436 str2int(后缀自动机)

题目链接：hdu 4436 str2int

题意：

给你n个字符串，每个字符串都是由数字构成，现在让你将这n个字符串所有的不重复子串构成的十进制数字加起来mod2012.

题解：

似乎这种不重复的子串问题，用后缀自动机都比较无脑搞。

首先将所有的串连起来，中间插个特殊字符，然后建立后缀自动机。

然后拓扑排序，从跟开始往下dp。

sum[v]=Σ(sum[x]*10+cnt[x]*j),其中cnt[x]表示有多少条路径能到x这个节点，转移为cnt[v]=Σcnt[x] (x能转移到v)，j为这个节点代表的数字。

#include<bits/stdc++.h>
#define mst(a,b) memset(a,b,sizeof(a))
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;

const int N=2e5+7,tyn=11,M=N*2,P=2012;
struct SAM{
    int tr[M][tyn],f[M],ml[M],ed,last,p,x,r,q;
    int b[M],d[M],cnt[M],sum[M];
    inline int gid(char x){return x-'0';}
    inline void nc(int s,int &p){
        ml[p=++ed]=s,f[ed]=cnt[ed]=sum[ed]=0,mst(tr[ed],0);
    }
    void clear(){ed=0,nc(0,last);}
    void add(int w){
        nc(ml[x=last]+1,p),last=p;
        while(x&&!tr[x][w])tr[x][w]=p,x=f[x];
        if(!x)f[p]=1;
        else if(ml[x]+1==ml[q=tr[x][w]])f[p]=q;
        else{
            nc(ml[x]+1,r),f[r]=f[q],f[p]=f[q]=r;
            memcpy(tr[r],tr[q],sizeof(tr[r]));
            while(x&&tr[x][w]==q)tr[x][w]=r,x=f[x];
        }
    }
    void upright(int mx=0){
        F(i,0,ed)d[i]=0;
        F(i,1,ed)d[mx<ml[i]?mx=ml[i]:ml[i]]++;
        F(i,1,mx)d[i]+=d[i-1];
        F(i,1,ed)b[d[ml[i]]--]=i;
    }
    void build(char *s){for(int i=1;s[i];i++)add(gid(s[i]));}
    
    void solve()
    {
        int ans=0;
        cnt[1]=1;
        F(i,1,ed)
        {
            int x=b[i],v;
            F(j,0,9)
            {
                v=tr[x][j];
                if((i==1&&j==0)||!v)continue;
                cnt[v]=(cnt[v]+cnt[x])%P;
                sum[v]=(sum[v]+sum[x]*10+cnt[x]*j)%P;
            }
            ans=(ans+sum[x])%P;
        }
        printf("%d
",ans);
    }
}sam;

char s[N];
int n;

int main(){
    while(~scanf("%d",&n))
    {
        sam.clear();
        F(i,1,n)scanf("%s",s+1),sam.build(s),sam.add(10);
        sam.upright(),sam.solve();
    }
    return 0;
}