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

题目描述

传送门

题解

今天胡策gty竟然原搬sdoi原题…
写完顺手过了样例，写暴力对拍竟然不出错！自己也吃了一惊，感觉稳得很
结果没处理前导0…GG

把所有的数字串丢到AC自动机里去
f[i][j][0/1][0/1]表示第i位数，匹配到AC自动机的第j个节点，现在是否填过不为0的数，是否卡上界的方案数
处理前导0的时候就是如果前面没有填过不为0的数那么一直停留在AC自动机的根节点
注意fail指针的下传

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
#define N 2005
#define Mod 1000000007

int n,m,sz,ans;
char num[N],s[N];
int ch[N][20],fail[N],is_end[N];
int tot,point[N],nxt[N],v[N];
int f[N][N][2][2];
queue <int> q;

void insert()
{
    int now=0,len=strlen(s);
    for (int i=0;i<len;++i)
    {
        int x=s[i]-'0';
        if (!ch[now][x]) ch[now][x]=++sz;
        now=ch[now][x];
    }
    is_end[now]=1;
}
void make_fail()
{
    while (!q.empty()) q.pop();
    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]);
        }
    }
}
void add(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
}
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()
{
    scanf("%s",num+1);n=strlen(num+1);
    scanf("%d",&m);
    for (int i=1;i<=m;++i)
    {
        scanf("%s",&s);
        insert();
    }
    make_fail();
    for (int i=1;i<=sz;++i) add(fail[i],i);
    dfs(0);
    ++f[1][0][0][0];
    for (int i=1;i<=num[1]-'0';++i)
        if (!is_end[ch[0][i]])
            ++f[1][ch[0][i]][1][i==num[1]-'0'?1:0];

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