【SDOI2014】数数

Description

求1~N中不含数字串集合S中的任意一个的数字的数量。
数字串可以有前导0，少于位数不用补0.
字符串集合中有M个串，字符集总长为L。
|N|<=1200,M<=100,L<=1500

Solution

AC自动机上DP。
设Fi,j,up,zero表示当前到第i位，AC自动机走到j这个节点，有没有爆上界，是否在算前导0.
其实最后一维可以不要，%各位大犇，蒟蒻不会。
随便乱搞就好了。

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define N 2005
#define mo 1000000007
#define ll long long
using namespace std;
struct note{
    int son[10],bz,next;
}trie[N];
int n,tot,d[N];
ll ans,f[N][N][2][2];
char s[N],st[N];
void makefail() {
    for(int i=0,j=1;i<j;) {
        int x=d[++i];
        fo(k,0,9) 
            if (trie[x].son[k]) {
                int y=trie[x].son[k];
                if (x) {
                    int z=trie[x].next;
                    while (z&&!trie[z].son[k]) z=trie[z].next;
                    trie[y].next=trie[z].son[k];
                }
                if (!trie[y].bz) trie[y].bz=trie[trie[y].next].bz;d[++j]=y;
            }
    }
}
int main() {
    freopen("count.in","r",stdin);
    freopen("count.out","w",stdout);
    scanf("%s",st+1);
    scanf("%d",&n);
    fo(i,1,n) {
        scanf("%s",s+1);int l=strlen(s+1),x=0;
        fo(j,1,l) {
            int y=s[j]-'0';
            if (!trie[x].son[y]) trie[x].son[y]=++tot;
            x=trie[x].son[y];
        }
        trie[x].bz=1;
    }
    makefail();
    int l=strlen(st+1);
    f[0][0][1][1]=1;
    fo(i,0,l-1)
        fo(j,0,tot)
            fo(up,0,1)
                fo(zero,0,1)
                    if (f[i][j][up][zero]) {
                        fo(k,0,9) {
                            int x=j;
                            if (!k&&zero) continue;
                            if (up&&k>st[i+1]-'0') continue;
                            while (x&&!trie[x].son[k]) x=trie[x].next;
                            x=trie[x].son[k];int y=up&&k==st[i+1]-'0';
                            if (trie[x].bz) continue;
                            f[i+1][x][y][0]=(f[i+1][x][y][0]+
                            f[i][j][up][zero])%mo;
                        }
                        if (zero) f[i+1][j][0][1]=(f[i+1][j][0][1]+
                        f[i][j][up][1]);
                    }
    fo(i,0,tot) fo(j,0,1) ans=(ans+f[l][i][j][0])%mo;
    printf("%lld",ans);
}