bzoj3530 [Sdoi2014]数数（AC自动机+数位DP）

bzoj3530 [Sdoi2014]数数

原题地址：http://www.lydsy.com/JudgeOnline/problem.php?id=3530

题意：
我们称一个正整数N是幸运数，当且仅当它的十进制表示中不包含数字串集合S中任意一个元素作为其子串。例如当S=(22，333，0233)时，233是幸运数，2333、20233、3223不是幸运数。
给定N和S，计算不大于N的幸运数个数。

数据范围
下表中l表示N的长度，L表示S中所有串长度之和。
1 < =l < =1200 , 1 < =M < =100 ,1 < =L < =1500

题解：
要求”不大于”，想到数位DP。
然后这个”不包含子串”需要AC自动机。
用的自己一般的数位dp打法：
dp[i][j][1/0][1/0]表示，当前第几位，AC自动机上何处，有没有lim，有没有前导零。

注意：
给出的串可能有前导零，所以当目前都是前导零时，不能走到ch[i][0]而只能留在根

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int N=1510;
const int mod=1000000007;
queue<int> Q;
int n=0,m,ch[N][10],fail[N],isw[N],root=0,tail=0,num[N],dp[N][N][2][2];
char s[N];
void insert()
{
    int len=strlen(s); int tmp=0;
    for(int i=0;i<len;i++)
    {
        int c=s[i]-'0';
        if(!ch[tmp][c]) ch[tmp][c]=++tail;
        tmp=ch[tmp][c];
    }
    isw[tmp]=1;
}
void getfail()
{
    for(int i=0;i<10;i++) if(ch[root][i]) Q.push(ch[root][i]);
    while(!Q.empty())
    {
        int tmp=Q.front(); Q.pop();
        for(int i=0;i<10;i++)
        {
            if(!ch[tmp][i]) {ch[tmp][i]=ch[fail[tmp]][i];continue;}
            int u=ch[tmp][i];
            fail[u]=ch[fail[tmp]][i];
            isw[u]|=isw[fail[u]];
            Q.push(u);
        }
    }
}
int main()
{
    scanf("%s%d",s,&m);
    n=strlen(s); for(int i=0;i<n;i++) num[i]=s[i]-'0';
    for(int i=1;i<=m;i++) {scanf("%s",s); insert();}
    getfail();
    dp[0][root][1][1]=1;
    for(int i=0;i<n;i++)
    for(int j=0;j<=tail;j++)
    for(int lim=0;lim<=1;lim++)
    for(int z=0;z<=1;z++)
    if(dp[i][j][lim][z])
    {
        int top=lim?num[i]:9;
        for(int c=0;c<=top;c++)
        {
            if(isw[ch[j][c]]) continue;
            int nlim=(lim&&c==top); int nz=(z&&c==0);
            if(!nz) dp[i+1][ch[j][c]][nlim][nz]=(dp[i+1][ch[j][c]][nlim][nz]+dp[i][j][lim][z])%mod;
            else dp[i+1][root][nlim][nz]=(dp[i+1][root][nlim][nz]+dp[i][j][lim][z])%mod;
        }
    }
    int ret=0;
    for(int i=0;i<=tail;i++)
    for(int lim=0;lim<=1;lim++) ret=(ret+dp[n][i][lim][0])%mod;
    printf("%d\n",ret);
    return 0;
}