【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=0,sign=1; char ch=getchar();
    while(ch<'0'||ch>'9'){ if (ch=='-') sign=-1; ch=getchar();}
    while(ch>='0'&&ch<='9'){ v=v*10+ch-'0'; ch=getchar();}
    return v*sign;
}
const int N=2015,INF=~0u>>2,MOD=1e9+7;
typedef long long LL;
/******************tamplate*********************/
int n,m,cnt=1,a[N];
struct Trie{
    int ch[10],fail;
    bool sign;
}T[N*10];
char s[N],s1[N];
void ins(){
    scanf("%s",s1);
    int x=1,y;
    rep(i,strlen(s1)){
        y=s1[i]-'0';
        if (!T[x].ch[y]) T[x].ch[y]=++cnt;
        x=T[x].ch[y];
    }
    T[x].sign=1;
}
int Q[N];
void make_fail(){
    int l=0,r=-1,j;
    Q[++r]=1;
    while(l<=r){
        int x=Q[l++],y;
        T[x].sign|=T[T[x].fail].sign;
        rep(i,10){
            j=T[x].fail;
            while(j && T[j].ch[i]==0) j=T[j].fail;
            if (T[x].ch[i]){
                y=T[x].ch[i];
                T[y].fail=j ? T[j].ch[i] : 1;
                Q[++r]=y;
            }else T[x].ch[i]=j ? T[j].ch[i] : 1;
        }
    }
}
int f[N][N][2];
int main(){
#ifndef ONLINE_JUDGE
    freopen("3530.in","r",stdin);
    freopen("3530.out","w",stdout);
#endif
    scanf("%s",s);
    n=strlen(s);
    F(i,1,n) a[i]=s[i-1]-'0';
    m=getint(); cnt=1;
    F(i,0,9) T[1].ch[i]=++cnt;
    F(i,1,m) ins();
    make_fail();
 
    F(i,1,a[1]) 
        if(!T[T[1].ch[i]].sign) f[1][T[1].ch[i]][i==a[1]]=1;
    F(i,1,n-1) F(j,1,cnt){
        F(k,0,a[i+1]) if(!T[T[j].ch[k]].sign)
            (f[i+1][T[j].ch[k]][k==a[i+1]]+=f[i][j][1])%=MOD;
        rep(k,10) if(!T[T[j].ch[k]].sign)
            (f[i+1][T[j].ch[k]][0]+=f[i][j][0])%=MOD;
    }
    int ans=0;
    F(i,1,cnt) (ans+=f[n][i][0])%=MOD,(ans+=f[n][i][1])%=MOD;
 
    memset(f,0,sizeof f);
    F(i,1,9) if (!T[T[1].ch[i]].sign) f[1][T[1].ch[i]][0]=1;
    F(i,1,n-2) F(j,1,cnt)
        rep(k,10) if(!T[T[j].ch[k]].sign)
            (f[i+1][T[j].ch[k]][0]+=f[i][j][0])%=MOD;
    F(i,1,n-1) F(j,1,cnt)
        (ans+=f[i][j][0])%=MOD;
    printf("%d
",ans);
    return 0;
}

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]