hdu4436 str2int（后缀自动机）

题意：

给若干个数字串,问这些串本质不同的子串的十进制之和.
答案对2012取模。

数据范围：串总长度<=1e5

解法：

将所有数字串插入SAM,插入新串前last置为1,

用p[x]表示到达状态x的方案数
用sum[x]表示状态x对答案的贡献
一开始p[root]=1

假设有转移边x->v,转移数字为j
那么p[v]+=p[x],sum[v]+=sum[x]*10+j*p[x]

最后答案为所有状态的sum[i]的和

注意:不允许前导零,前导零不转移

code：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxm=2e6+5;
const int mod=2012;
char s[maxm];
int n;
struct SAM{
    int ch[maxm][26];
    int fa[maxm],l[maxm];//l[]是等价类的最长字符串长度len
    int last=1,tot=1;//tot是节点数量
    void add(int c){
        int p=last,np=++tot;last=np;l[np]=l[p]+1;
        for(;p&&!ch[p][c];p=fa[p])ch[p][c]=np;
        if(!p)fa[np]=1;
        else{
            int q=ch[p][c];
            if(l[p]+1==l[q])fa[np]=q;
            else{
                int nq=++tot;l[nq]=l[p]+1;
                memcpy(ch[nq],ch[q],sizeof ch[q]);
                fa[nq]=fa[q];fa[q]=fa[np]=nq;
                for(;ch[p][c]==q;p=fa[p])ch[p][c]=nq;
            }
        }
    }
    void init(){
        for(int i=1;i<=tot;i++){
            memset(ch[i],0,sizeof ch[i]);
            fa[i]=l[i]=0;
        }
        last=tot=1;
    }
    //本题
    int cnt[maxm];
    int idx[maxm];
    int p[maxm];//p[x]表示到达状态x的方案数
    int sum[maxm];//sum[x]表示状态x对答案的贡献
    ll solve(){
        //clear
        for(int i=0;i<=tot;i++){
            cnt[l[i]]=0;
            sum[i]=0;
            p[i]=0;
        }
        //基数排序
        for(int i=1;i<=tot;i++)cnt[l[i]]++;
        for(int i=1;i<=tot;i++)cnt[i]+=cnt[i-1];
        for(int i=1;i<=tot;i++)idx[cnt[l[i]]--]=i;
        //
        p[1]=1;
        //
        for(int i=1;i<=tot;i++){
            int x=idx[i];
            for(int j=0;j<10;j++){
                if(ch[x][j]){
                    if(i==1&&j==0)continue;//前导零
                    int v=ch[x][j];
                    p[v]+=p[x];
                    sum[v]+=sum[x]*10+j*p[x];
                    sum[v]%=mod;
                }
            }
        }
        ll ans=0;
        for(int i=1;i<=tot;i++){
            ans+=sum[i];
            ans%=mod;
        }
        return ans;
    }
    //
}S;
signed main(){
    while(scanf("%d",&n)!=EOF){
        S.init();
        for(int i=1;i<=n;i++){
            scanf("%s",s+1);
            int len=strlen(s+1);
            for(int j=1;j<=len;j++){
                S.add(s[j]-'0');
            }
            S.last=1;
        }
        ll ans=S.solve();
        cout<<ans<<endl;
    }
    return 0;
}

---