【BZOJ】【3530】【SDOI2014】数数

AC自动机/数位DP

---

　　orz zyf

　　好题啊= =同时加深了我对AC自动机（这个应该可以叫Trie图了吧……出边补全！）和数位DP的理解……不过不能自己写出来还真是弱……

 /**************************************************************
     Problem: 3530
     User: Tunix
     Language: C++
     Result: Accepted
     Time:1008 ms
     Memory:33956 kb
 ****************************************************************/

 //BZOJ 3530
 #include<vector>
 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<iostream>
 #include<algorithm>
 #define rep(i,n) for(int i=0;i<n;++i)
 #define F(i,j,n) for(int i=j;i<=n;++i)
 #define D(i,j,n) for(int i=j;i>=n;--i)
 #define pb push_back
 using namespace std;
 inline int getint(){
     int v=,sign=; char ch=getchar();
     while(ch<''||ch>''){ if (ch=='-') sign=-; ch=getchar();}
     while(ch>=''&&ch<=''){ v=v*+ch-''; ch=getchar();}
     return v*sign;
 }
 const int N=,INF=~0u>>,MOD=1e9+;
 typedef long long LL;
 /******************tamplate*********************/
 int n,m,cnt=,a[N];
 struct Trie{
     int ch[],fail;
     bool sign;
 }T[N*];
 char s[N],s1[N];
 void ins(){
     scanf("%s",s1);
     int x=,y;
     rep(i,strlen(s1)){
         y=s1[i]-'';
         if (!T[x].ch[y]) T[x].ch[y]=++cnt;
         x=T[x].ch[y];
     }
     T[x].sign=;
 }
 int Q[N];
 void make_fail(){
     int l=,r=-,j;
     Q[++r]=;
     while(l<=r){
         int x=Q[l++],y;
         T[x].sign|=T[T[x].fail].sign;
         rep(i,){
             j=T[x].fail;
             while(j && T[j].ch[i]==) j=T[j].fail;
             if (T[x].ch[i]){
                 y=T[x].ch[i];
                 T[y].fail=j ? T[j].ch[i] : ;
                 Q[++r]=y;
             }else T[x].ch[i]=j ? T[j].ch[i] : ;
         }
     }
 }
 int f[N][N][];
 int main(){
 #ifndef ONLINE_JUDGE
     freopen("3530.in","r",stdin);
     freopen("3530.out","w",stdout);
 #endif
     scanf("%s",s);
     n=strlen(s);
     F(i,,n) a[i]=s[i-]-'';
     m=getint(); cnt=;
     F(i,,) T[].ch[i]=++cnt;
     F(i,,m) ins();
     make_fail();

     F(i,,a[])
         if(!T[T[].ch[i]].sign) f[][T[].ch[i]][i==a[]]=;
     F(i,,n-) F(j,,cnt){
         F(k,,a[i+]) if(!T[T[j].ch[k]].sign)
             (f[i+][T[j].ch[k]][k==a[i+]]+=f[i][j][])%=MOD;
         rep(k,) if(!T[T[j].ch[k]].sign)
             (f[i+][T[j].ch[k]][]+=f[i][j][])%=MOD;
     }
     int ans=;
     F(i,,cnt) (ans+=f[n][i][])%=MOD,(ans+=f[n][i][])%=MOD;

     memset(f,,sizeof f);
     F(i,,) if (!T[T[].ch[i]].sign) f[][T[].ch[i]][]=;
     F(i,,n-) F(j,,cnt)
         rep(k,) if(!T[T[j].ch[k]].sign)
             (f[i+][T[j].ch[k]][]+=f[i][j][])%=MOD;
     F(i,,n-) F(j,,cnt)
         (ans+=f[i][j][])%=MOD;
     printf("%d\n",ans);
     return ;
 }

3530: [Sdoi2014]数数

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 404  Solved: 241
[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

[Submit][Status][Discuss]