【解题报告】HDU4436-str2int 广义SAM

考虑后缀自动机，首先先用个siz数组来记录到这个状态的子串的次数，用sum数组来表示到这个状态的和。于是我们就能找到个公式就是对于数j来说，sum[mp(i,j)]+=10sum[i]+j*siz[i]，意思就是到状态i有siz[i]个子串，走数j的话，那就相当于当前的数乘10然后再加上现在的数。然后这个是要用拓扑序来走的，从小到大，走的时候同时还要求siz，siz[mp(i,j)]+=siz[i]。可以随便找个SAM试试就明白。

然后由于有前导0的存在，所以从根节点开始走的话不能走0，为什么的话用一个SAM试试就明白。

AC代码：

#include <bits/stdc++.h>
#define Re register int
using namespace std;
typedef long long LL;
const int N=2e5+5;
const int mod=2012;
int len;char s[N];
struct SAM{
    int cnt,fa[N],mlen[N],mp[N][10];
    LL sum[N],siz[N];
    int buc[N],oid[N];
    //fa[i]: 后缀链接
    //mp[i]: 状态转移数组
    SAM(){cnt=1;}//根初始化为1
    void clear()
    {
    	memset(mp,0,sizeof(mp));
    	memset(fa,0,sizeof(fa));
    	memset(mlen,0,sizeof(mlen));
    	memset(sum,0,sizeof(sum));
    	memset(buc,0,sizeof(buc));
    	memset(oid,0,sizeof(oid));
    	memset(siz,0,sizeof(siz));
    	cnt=1;
	} 
    inline int insert(Re ch,Re last){
        if(mp[last][ch]){
            Re p=last,x=mp[p][ch];
            if(mlen[p]+1==mlen[x])return x;//即最初的特判1
            else{
                Re y=++cnt;mlen[y]=mlen[p]+1;
                for(Re i=0;i<10;++i)mp[y][i]=mp[x][i];
                while(p&&mp[p][ch]==x)mp[p][ch]=y,p=fa[p];
                fa[y]=fa[x],fa[x]=y;
                return y;//即最初的特判2
            }
        }
        Re z=++cnt,p=last;mlen[z]=mlen[last]+1;
        while(p&&!mp[p][ch])mp[p][ch]=z,p=fa[p];
        if(!p)fa[z]=1;
        else{
            Re x=mp[p][ch];
            if(mlen[p]+1==mlen[x])fa[z]=x;
            else{
                Re y=++cnt;mlen[y]=mlen[p]+1;
                for(Re i=0;i<10;++i)mp[y][i]=mp[x][i];
                while(p&&mp[p][ch]==x)mp[p][ch]=y,p=fa[p];
                fa[y]=fa[x],fa[z]=fa[x]=y;
            }
        }
        return z;
    }
    void work()
    {
    	for(int i=1;i<=cnt;i++) buc[mlen[i]]++;
		for(int i=1;i<=cnt;i++) buc[i]+=buc[i-1];
		for(int i=cnt;i>=1;i--) oid[buc[mlen[i]]--]=i;
		siz[1]=1;
		for(int i=1;i<=cnt;i++){
			int pos=oid[i];
			for(int j=0;j<10;j++){
				if(!mp[pos][j]) continue;
				if(pos==1&&j==0) continue;
				int nxt=mp[pos][j];
				siz[nxt]+=siz[pos];
				sum[nxt]+=10*sum[pos]+j*siz[pos];
				sum[nxt]%=mod;
			}
		}
		LL ans=0;
		for(int i=1;i<=cnt;i++){
			ans+=sum[i];
			ans%=mod;
		}
		printf("%lld\n",ans);
	}
    void solve()
    {
    	int n;
    	while(scanf("%d",&n)!=EOF){
    		clear();
    		for(int i=1;i<=n;i++){
    			scanf("%s",s+1);
    			len=strlen(s+1);
    			int last=1;
    			for(int j=1;j<=len;j++){
    				last=insert(s[j]-'0',last);
				}
			}
			work();
		}
	}
}sam;
int main()
{
	sam.solve();
}