[BZOJ3530][SDOI2014]数数（AC自动机+数位dp）

题目：

我是超链接

题解：

我们先把数字串安置在AC自动机上
f[i][j][0/1][0/1]表示第i位数，匹配到AC自动机的第j个节点，现在是否填过不为0的数，是否卡上界的方案数
处理前导0的时候就是如果前面没有填过不为0的数那么一直停留在AC自动机的根节点
注意fail指针的下传
这个前导零是个什么情况呢？N=111 00∈S 那么001也是合法的啊
这个第四维表示是否卡上界是数位dp的常见姿势了

代码：

#include <queue>
#include <cstdio>
#include <cstring>
using namespace std;
const int mod=1e9+7;
const int N=2005;
int tot,nxt[N],point[N],v[N],ch[N][12],cnt,fail[N],f[1205][1505][2][2];char st[N],s[N];bool is_end[N];
void addline(int x,int y){++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;}
void insert()
{
    scanf("%s",s);
    int now=0;int l=strlen(s);
    for (int i=0;i<l;i++)
    {
        int x=s[i]-'0';
        if (!ch[now][x]) ch[now][x]=++cnt;
        now=ch[now][x];
    }
    is_end[now]=1;
}
void sp()
{
    queue<int>q;
    for (int i=0;i<=9;i++)
      if (ch[0][i]) q.push(ch[0][i]);
    while (!q.empty())
    {
        int now=q.front(); q.pop();
        for (int i=0;i<=9;i++)
        {
            if (!ch[now][i]) 
            {
                ch[now][i]=ch[fail[now]][i];
                continue;
            }
            fail[ch[now][i]]=ch[fail[now]][i];
            q.push(ch[now][i]);
        }
    }
    for (int i=1;i<=cnt;i++) addline(fail[i],i);
}
void dfs(int x)
{
    for (int i=point[x];i;i=nxt[i])
      is_end[v[i]]|=is_end[x],dfs(v[i]);
}
int main()
{
    int n,m;scanf("%s",st+1); n=strlen(st+1);
    scanf("%d",&m);
    for (int i=1;i<=m;i++) insert();
    sp(); dfs(0);
    //f[i][j][0/1][0/1]第i位匹配到AC自动机的第j位，是否匹配过非0的数，是否达到上界 
    ++f[1][0][0][0];
    for (int i=1;i<=st[1]-'0';i++)
      if (!is_end[ch[0][i]]) f[1][ch[0][i]][1][i==st[1]-'0'?1:0]++;

    for (int i=1;i<=n;i++)
      for (int j=0;j<=cnt;j++)
        for (int k=0;k<=1;k++)
          for (int l=0;l<=1;l++)
          {
            int up=9,s=st[i+1]-'0';
            if (l) up=s;
            for (int x=0;x<=up;x++)
            {
                if (k==0 && x==0) 
                {
                    f[i+1][0][0][l&&(x==st[i+1]-'0')?1:0]+=f[i][j][k][l];
                    f[i+1][0][0][l&&(x==st[i+1]-'0')?1:0]%=mod;
                    continue;
                }
                if (is_end[ch[j][x]]) continue;
                f[i+1][ch[j][x]][1][l&&(x==st[i+1]-'0')?1:0]+=f[i][j][k][l];
                f[i+1][ch[j][x]][1][l&&(x==st[i+1]-'0')?1:0]%=mod;
            }
          }
    int ans=0;
    for (int i=0;i<=cnt;i++)
      for (int j=0;j<=1;j++)
        if (!is_end[i]) ans=(ans+f[n][i][1][j])%mod;
    printf("%d",ans);
}