考虑先建出集合S的AC自动机,并把它补成Trie图(是叫这个吗?),并在AC自动机标记上S集合中每个数字的结束位置。把这个标记沿fail树下传之后,一个串只要在AC自动机上没有走到标记点,就说明它是幸运数.
所以考虑数位DP,设
f[i][j][0/1]
表示从高往低填到第
i
位,跑到AC自动机上
考虑到不能有前导零,所以填每一位的时候可以把根节点单独转移一次1~9。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#define up(y,x) x=((x)+(y))%mod
using namespace std;
const int mod=1000000007;
char lim[1210],cs[1510];
int trie[1510][10],fa[1510],dl[1510],n,m,tot,f[1210][1510][2];
bool pl[1510];
void ins()
{
int p=0;
for(int l=strlen(cs+1),i=1;i<=l;i++)
{
if(!trie[p][cs[i]-'0']) trie[p][cs[i]-'0']=++tot;
p=trie[p][cs[i]-'0'];
}
pl[p]=1;
}
void getac()
{
for(int head=1,tail=1,v=0;head<=tail;v=dl[++head])
for(int c=0;c<=9;c++)
if(trie[v][c])
{
int p=fa[v];
for(;p&&!trie[p][c];p=fa[p]);
if(trie[p][c]!=trie[v][c]) fa[trie[v][c]]=trie[p][c];
pl[trie[v][c]]|=pl[trie[p][c]];
dl[++tail]=trie[v][c];
}
else trie[v][c]=trie[fa[v]][c];
}
void solve()
{
for(int i=1;i<=n;i++)
{
for(int c=1,v=trie[0][c];c<=9;v=trie[0][++c])
{
if(i!=1||c<lim[i]) up(1,f[i][v][0]);
else if(c==lim[i]) up(1,f[i][v][1]);
}
for(int j=0;j<=tot;j++)
if(!pl[j])
for(int c=0,v=trie[j][c];c<=9;v=trie[j][++c])
{
up(f[i-1][j][0],f[i][v][0]);
if(c<lim[i]) up(f[i-1][j][1],f[i][v][0]);
else if(c==lim[i]) up(f[i-1][j][1],f[i][v][1]);
}
}
int ans=0;
for(int j=0;j<=tot;j++)
if(!pl[j]) up(f[n][j][0],ans),up(f[n][j][1],ans);
printf("%d",ans);
}
int main()
{
scanf("%s",lim+1);
n=strlen(lim+1);
for(int i=1;i<=n;i++)
lim[i]-='0';
scanf("%d",&m);
for(int i=1;i<=m;i++)
{
scanf("%s",cs+1);
ins();
}
getac();
solve();
return 0;
}