bzoj [Sdoi2014]数数 AC自动机上dp

[Sdoi2014]数数

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 1264  Solved: 636
[Submit][Status][Discuss]

Description

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

Input

输入的第一行包含整数N。
    接下来一行一个整数M，表示S中元素的数量。
    接下来M行，每行一个数字串，表示S中的一个元素。

Output

输出一行一个整数，表示答案模109+7的值。

Sample Input

20
3
2
3
14

Sample Output

14

HINT

下表中l表示N的长度，L表示S中所有串长度之和。

1 < =l < =1200 , 1 < =M < =100 ,1 < =L < =1500

Source

Round 1 day 1

AC自动机建好后f[i][j][k]表示i这个点j长度k=0表示小于，1表示等于

这样转移很好转移 IL的复杂度随便跑，

虑到不能有前导零，所以填每一位的时候可以把根节点单独转移一次1~9

 #pragma GCC optimzie(2)
 #pragma G++ optimize(2)
 #include<cstring>
 #include<cmath>
 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 #include<queue>

 #define ll long long
 #define N 2007
 #define mod 1000000007
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(!isdigit(ch)){if(ch=='-')f=-;ch=getchar();}
     while(isdigit(ch)){x=(x<<)+(x<<)+ch-'';ch=getchar();}
     return x*f;
 }

 int m,len,l;
 ll ans;
 struct Node
 {
     int c[],suf;
     bool flag;
 }trie[N];
 int cnt=,rt=;
 ll f[][][];
 char ch[N],n[N];

 void insert()
 {
     int now=rt;
     for (int i=;i<=len;i++)
         if(trie[now].c[ch[i]-''])now=trie[now].c[ch[i]-''];
         else
         {
             trie[now].c[ch[i]-'']=++cnt;
             now=cnt;
         }
     trie[now].flag=true;
 }
 void make_AC()
 {
     for (int i=;i<;i++)
         trie[].c[i]=;
     queue<int>q;q.push();
     trie[].suf=;
     while(!q.empty())
     {
         int u=q.front();q.pop();
         for (int i=;i<;i++)
             if(trie[u].c[i])
             {
                 trie[trie[u].c[i]].suf=trie[trie[u].suf].c[i];
                 if(trie[trie[trie[u].c[i]].suf].flag)trie[trie[u].c[i]].flag=true;
                 q.push(trie[u].c[i]);
             }
             else trie[u].c[i]=trie[trie[u].suf].c[i];
     }
 }
 void solve()
 {
     /*f[0][1][1]=1;
     for (int i=0;i<l;i++)
         for (int j=1;j<=cnt;j++)
         {
             if (f[i][j][1]&&!trie[j].flag)
             {
                 for (int k=0;k<10;k++)
                     if (n[i+1]-'0'>k)(f[i+1][trie[j].c[k]][0]+=f[i][j][1])%=mod;
                     else if(n[i+1]-'0'==k)(f[i+1][trie[j].c[k]][1]+=f[i][j][1])%=mod;
             }
             if(f[i][j][0]&&!trie[j].flag)
             {
                 for (int k=0;k<10;k++)
                     (f[i+1][trie[j].c[k]][0]+=f[i][j][0])%=mod;
             }
         }*/
     for (int i=;i<=l;i++)
     {
         for (int z=,v=trie[].c[z];z<=;v=trie[].c[++z])
         {
             if(i!=||z<n[i]-'')(f[i][v][]+=)%=mod;
             else if(z==n[i]-'')(f[i][v][]+=)%=mod;
         }
         for (int j=;j<=cnt;j++)
             if(!trie[j].flag)
                 for (int c=,v=trie[j].c[c];c<=;v=trie[j].c[++c])
                 {
                     (f[i][v][]+=f[i-][j][])%=mod;
                     if(c<n[i]-'')(f[i][v][]+=f[i-][j][])%=mod;
                     else if(c==n[i]-'')(f[i][v][]+=f[i-][j][])%=mod;
                 }
     }

     for (int i=;i<=cnt;i++)
         if(!trie[i].flag)(ans+=f[l][i][]+f[l][i][])%=mod;
     printf("%lld",ans);
 }
 int main()
 {
     scanf("%s",n+);l=strlen(n+);
     m=read();
     for (int i=;i<=m;i++)
     {
         scanf("%s",ch+),len=strlen(ch+);
         insert();
     }
     make_AC(),solve();
 }