BUPT Confusing Problem(自动机+DP)

题目链接：http://acm.bupt.edu.cn/onlinejudge/newoj/showProblem/show_problem.php?problem_id=652

题意：给定数字A和B，问区间[L,R]之间有多少个数字不包含0且至少包含数字A或B中的一个?

思路：用A和B建立自动机。f[dep][id][allZero][ok]表示深度dep、节点id、之前是否全0、是否包含A或B的个数。

struct node

{

    int next[10],fail,flag;



    void init()

    {

        clr(next,0);

        fail=0;

        flag=0;

    }

};



node a[N];

int e;

i64 n,m;





void insert(char s[])

{

    int i,k,p=0;

    for(i=0;s[i];i++)

    {

        k=s[i]-'0';

        if(a[p].next[k]==0)

        {

            a[e].init();

            a[p].next[k]=e++;

        }

        p=a[p].next[k];

    }

    a[p].flag=1;

}









queue<int> Q;



void build()

{

    int i,j,k,p,q;

    FOR0(i,10) if(a[0].next[i]) Q.push(a[0].next[i]);

    while(!Q.empty())

    {

        k=Q.front();

        Q.pop();

        for(i=0;i<10;i++)

        {

            if(a[k].next[i])

            {

                p=a[k].next[i];

                q=a[k].fail;

                Q.push(p);

                a[p].fail=a[q].next[i];

                a[p].flag|=a[a[p].fail].flag;

            }

            else

            {

                q=a[k].fail;

                a[k].next[i]=a[q].next[i];

            }

        }

    }

}





i64 f[20][205][2][2];

int b[20],bNum;





i64 DFS(int id,int dep,int flag,int ok,int allZero)

{

    if(dep==-1) return ok==1;

    if(!flag&&f[dep][id][allZero][ok]!=-1) return f[dep][id][allZero][ok];

    int R=flag?b[dep]:9;

    int i,x;

    i64 ans=0;

    if(allZero&&dep) ans+=DFS(id,dep-1,flag&&R==0,ok,allZero);

    FOR1(i,R)

    {

        x=a[a[id].next[i]].flag;

        ans+=DFS(a[id].next[i],dep-1,flag&&i==R,ok|x,0);

    }

    if(!flag) f[dep][id][allZero][ok]=ans;

    return ans;

}



i64 cal(i64 x)

{

    clr(f,-1);

    bNum=0;

    while(x)

    {

        b[bNum++]=x%10;

        x/=10;

    }

    return DFS(0,bNum-1,1,0,1);

}





char str[30];



int main()

{

    int C;

    RD(C);

    while(C--)

    {

        a[0].init();e=1;

        int i;

        RD(n,m);

        RD(str); insert(str);

        RD(str); insert(str);

        build();

        i64 ans=cal(m)-cal(n-1);

        PR(ans);

    }

    return 0;

}