HDU4436 后缀自动机+dp

拿到题的第一眼 咦 好熟悉

行云流水建个后缀自动机 在自动机上dfs一发 秒过样例 直接交题 然后T到心态爆炸

后来了解到这道题现场赛只有13个队出 金牌题？？？

不过毕竟是12年的区域赛 那个时候后缀自动机可能还没有现在这样普及

于是被迫思考dp写法 记得有一道题和这题dp思路很像 然后试了一发 果然如此 可以认为是蒙出来的QAQ

耗时150ms 但是题目给了3s是什么鬼？ 可能出这题的初衷是让用后缀数组做吧 然而并不会后缀数组啊

dp思路很明显了 代码应该一看就懂（看不懂就去多刷几道后缀自动机的模板题吧...）

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
const int mod=2012;
char s[maxn];
int ans;
int n;
struct Suffix_Automation
{
    int next[maxn*2][11],fa[maxn*2],l[maxn*2];
    int last,cnt,sum[maxn*2],num[maxn*2];
    int cntA[maxn*2],A[maxn*2];
    void init()
    {
        last=cnt=1;
        fa[1]=l[1]=0;
        memset(next[1],0,sizeof next[1]);
        memset(sum,0,sizeof sum);
        memset(num,0,sizeof num);
    }
    int inline newnode()
    {
        cnt++;
        memset(next[cnt],0,sizeof next[cnt]);
        fa[cnt]=l[cnt]=0;
        return cnt;
    }
    void add(int c)
    {
        int p=last;
        int np=newnode();
        l[np]=l[p]+1;
        last=np;
        while(p&&!next[p][c])
        {
            next[p][c]=np,p=fa[p];
        }
        if(!p)
            fa[np]=1;
        else
        {
            int q=next[p][c];
            if(l[q]==l[p]+1)
                fa[np]=q;
            else
            {
                int nq=++cnt;
                l[nq]=l[p]+1;
                memcpy(next[nq],next[q],sizeof next[q]);
                fa[nq]=fa[q];
                fa[np]=fa[q]=nq;
                while(next[p][c]==q)
                    next[p][c]=nq,p=fa[p];
            }
        }
    }
    void work()
    {
        memset(cntA,0,sizeof cntA);
        for(int i=1;i<=cnt;i++)
            cntA[l[i]]++;
        for(int i=1;i<=n;i++)
            cntA[i]+=cntA[i-1];
        for(int i=cnt;i>=1;i--)
            A[cntA[l[i]]--]=i;
    }
    void solve()
    {
        num[1]=1;
        for(int i=1;i<=cnt;i++)
        {
            int t=A[i];
            for(int j=0;j<10;j++)
            {
                if(t==1&&j==0)
                    continue;
                if(next[t][j])
                {
                    num[next[t][j]]=(num[next[t][j]]+num[t])%mod;
                    sum[next[t][j]]=(sum[next[t][j]]+sum[t]*10+num[t]*j)%mod;
                }
            }
        }
        for(int i=1;i<=cnt;i++)
            ans=(ans+sum[i])%mod;
    }
}sam;
int main()
{
    int all;
    while(scanf("%d",&all)!=EOF)
    {
        sam.init();
        ans=0;
        n=all;
        for(int i=1;i<=all;i++)
        {
            scanf("%s",s+1);
            int len=strlen(s+1);
            n+=len;
            for(int i=1;i<=len;i++)
                sam.add(s[i]-'0');
            sam.add(10);
        }
        sam.work();
        sam.solve();
        printf("%d\n",ans);
    }
    return 0;
}