[SDOI2014]数数------ac自动机+dfs

原题链接
这一题的dfs比较特殊，如果不用dfs，要从1到n，肯定会超时，所以这里巧妙的用了dfs，这里给了超内存的代码，如果要ac，可以把MX的数值开小一点。

先解释一下dfs后面那一个lim和zero是什么意思吧
lim是“是否最高有限制”，zero是“是否前面为0”
lim，如果有限制，那么就不能到达9了；
比如，27，第一次走‘0’，第二次走‘1’，都无限制，第二位可以0~9；
第三次走‘2’，有限制，第二位只可以走0~7；
有限制的前提是前面所有的位，均为有限制，也就是，前面所有的数字均为num数组里的数字；

zero是“前面是否全部为0
首先说为什么会引入一个zero，比如，1234，字符串是从1开始，而0234，字符串是从2开始，而与第一个无关。
如果zero为true，那么起点就是从当前这个数字开始，而与前面的数字全部无关。
如果要zero==true，那么必须要此数位的前面所有数字都是0，那么，这个数位的数字才可以为‘根’。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#define ll long long
using namespace std;
const int mx=1e3+9;
const int mod=1e9+7;
const int MX=4e5+9;
char s[mx];
int n,num[mx],cnt=1,t[MX][10],ed[MX],fail[MX],dp[MX][mx];

void inser(){
    int now=1;
    for( int i=0 ; s[i] ; i++ ){
        int c=s[i]-'0';
        if( !t[now][c] )
            t[now][c]=++cnt;
        now=t[now][c];
    }
    ed[now]=1;
}

void build(){
    queue<int> que;
    for( int i=0 ; i<=9 ; i++ ){
        if( !t[1][i] )
            t[1][i]=1;
        else{
            fail[t[1][i]]=1;
            que.push(t[1][i]);
        }
    }
    while( !que.empty() ){
        int now=que.front();
        que.pop();
        for( int i=0 ; i<10 ; i++ ){
            if( !t[now][i] )
                t[now][i]=t[fail[now]][i];
            else{
                fail[t[now][i]]=t[fail[now]][i];
                que.push(t[now][i]);
                ed[t[now][i]]=ed[t[now][i]]|ed[fail[t[now][i]]];   // 因为fail树的儿子一定包揽父亲的后缀，
            }                             //所以当父亲拥有s字符串的时候，它的所有儿子必有。
        }
    }
}
// 着重说一下这个dfs
int dfs(int st,int len,bool lim,bool zero){
    if( !lim && !zero && dp[st][len]!=-1 )    
        return dp[st][len];     // 如果曾经走过，你们直接返回（剪枝）
    if( len==0 )
        return zero?0:1;   // 如果走完了，如果zero为true，那么就是‘无’，否则就找到一个符合的数字
    int up=lim?num[len]:9;
    ll res=0;
    for( int i=0 ; i<=up ; i++ ){
        int cur=(zero&&i==0)?st:t[st][i];
        if( ed[cur] )
            continue;   // 如果这个数字里面包括了s的字符串，那么ed必为1，且此ed的子树必全部为1，所以我们不走它
        else
            res=(res+dfs(cur,len-1,i==up&&lim,zero==1&&i==0))%mod;
    }
    if( !lim && !zero )
        dp[st][len]=res;
    return res;
}

int main()
{
    freopen("input.txt","r",stdin);
    memset(fail,0,sizeof(fail));
    memset(dp,-1,sizeof(dp));
    scanf("%s",s);
    n=strlen(s);
    for( int i=0 ; i<n ; i++ )
        num[n-i]=s[i]-'0';
    int m;
    scanf("%d",&m);
    while( m-- ){
        scanf("%s",s);
        inser();
    }                   // 建一颗ac自动机树
    build();
    printf("%d\n",dfs(1,n,true,true));  
    return 0;
}