牛客8827 D.Chinese Valentine‘s Day（SAM）

题意：

给定n个数字串，求所有本质不同的子串的数位和。
答案对1e9+7取模。

数据范围：n<=1e3，总长度不超过1e6

解法：

将所有数字串插入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 int long long
const int maxm=2e6+5;
const int mod=1e9+7;
char s[maxm];
int n;
struct SAM{
    int ch[maxm][10];
    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];
    int sum[maxm];
    int 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]){
                    int v=ch[x][j];
                    p[v]+=p[x];
                    sum[v]+=sum[x]*10+j*p[x];
                    sum[v]%=mod;
                }
            }
        }
        int ans=0;
        for(int i=1;i<=tot;i++){
            ans+=sum[i];
            ans%=mod;
        }
        return ans;
    }
}T;
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    T.init();
    cin>>n;
    for(int i=1;i<=n;i++){
        string s;cin>>s;
        for(auto x:s){
            T.add(x-'0');
        }
        T.last=1;
    }
    int ans=T.solve();
    cout<<ans<<endl;
    return 0;
}

---