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

[Submit][Status][Discuss]

05-06 09:34