BZOJ 3530 [SDOI2014]数数 (Trie图/AC自动机+数位DP)

题目大意：略

裸的AC自动机+数位DP吧...

定义f[i][x][0/1]表示已经匹配到了第i位，当前位置是x，0表示没到上限，1到上限，此时数是数量

然而会出现虚拟前导零，即前几位没有数字的情况，实际上是在0号节点原地打转，所以多加一维状态，再额外讨论第1位就行了

#include <cmath>
#include <queue>
#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define NN 1210
#define MM 1510
#define ll long long
#define dd double
#define uint unsigned int
#define mod 1000000007
#define idx(X) (X-'0')
#define eps (1e-9)
using namespace std;

int n,m,cte;
int head[NN];
struct Edge{int to,nxt;}edge[NN*];
void ae(int u,int v){
    cte++;edge[cte].nxt=head[u];
    head[u]=cte,edge[cte].to=v;
}
uint f[NN][MM][];
int ma[NN];

namespace AC{
int ch[NN][],fail[NN],tot,ed[NN];
void Build_Trie(char *str,int len)
{
    int x=;
    for(int i=;i<=len;i++){
        if(!ch[x][idx(str[i])])
            ch[x][idx(str[i])]=++tot;
        x=ch[x][idx(str[i])];
    }ed[x]=;
}
void Build_Fail()
{
    queue<int>q;
    for(int i=;i<=;i++)
        if(ch[][i]) q.push(ch[][i]);
    while(!q.empty())
    {
        int x=q.front();q.pop();
        ae(fail[x],x);
        for(int i=;i<=;i++)
        {
            if(ch[x][i]){
                fail[ch[x][i]]=ch[fail[x]][i];
                q.push(ch[x][i]);
            }else{
                ch[x][i]=ch[fail[x]][i];
            }
        }
    }
    q.push();
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(int j=head[x];j;j=edge[j].nxt){
            int v=edge[j].to;
            ed[v]|=ed[x];
            q.push(v);
        }
    }
}
void solve()
{
    f[][][]++;
    for(int j=;j<ma[];j++){
        if(!ed[ch[][j]])
        f[][ch[][j]][]++;}
    if(!ed[ch[][ma[]]])
        f[][ch[][ma[]]][]++;
    for(int i=;i<n;i++)
    {
        f[i+][][]=;
        int x=;
        //nolimited
        for(int j=;j<=;j++){
            if(!ed[ch[x][j]])
            (f[i+][ch[x][j]][]+=f[i][x][]+((j!=)?f[i][x][]:))%=mod;
        }
        //limited
        for(int j=;j<ma[i+];j++)
            if(!ed[ch[x][j]])
            (f[i+][ch[x][j]][]+=f[i][x][])%=mod;
        if(!ed[ch[x][ma[i+]]])
            (f[i+][ch[x][ma[i+]]][]+=f[i][x][])%=mod;
        for(x=;x<=tot;x++)
        {
            //nolimited
            for(int j=;j<=;j++){
                if(!ed[ch[x][j]])
                (f[i+][ch[x][j]][]+=f[i][x][])%=mod;
            }
            //limited
            for(int j=;j<ma[i+];j++)
                if(!ed[ch[x][j]])
                (f[i+][ch[x][j]][]+=f[i][x][])%=mod;
            if(!ed[ch[x][ma[i+]]])
                (f[i+][ch[x][ma[i+]]][]+=f[i][x][])%=mod;
            //(f[i+1][ch[x][0]][2]+=f[i][x][2])%=mod;
        }
    }
    uint ans=;
    for(int x=;x<=tot;x++)
        (ans+=f[n][x][]+f[n][x][])%=mod;
    printf("%u\n",ans);
}
};
char str[NN];

int main()
{
    //freopen("t1.in","r",stdin);
    scanf("%s",str+);
    n=strlen(str+);
    for(int i=;i<=n;i++)
        ma[i]=idx(str[i]);
    scanf("%d",&m);
    for(int i=;i<=m;i++){
        scanf("%s",str+);
        int len=strlen(str+);
        AC::Build_Trie(str,len);
    }AC::Build_Fail();
    AC::solve();
    return ;
}