【HDU3530】 [Sdoi2014]数数 （AC自动机+数位DP）

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
#define Maxn 1600
#define Maxl 1600
#define Mod 1000000007

struct node
{
int fail,mark;
int son[30];
}t[Maxn];int tot;

int m,sl;

void upd(int x)
{
t[x].mark=0;
memset(t[x].son,0,sizeof(t[x].son));
}

char s[Maxl];
char ss[Maxn];
void read_trie()
{
scanf("%s",s+1);
int len=strlen(s+1);
int now=0;
for(int i=1;i<=len;i++)
{
int ind=s[i]-'0'+1;
if(!t[now].son[ind])
{
t[now].son[ind]=++tot;
upd(tot);
}
now=t[now].son[ind];
if(i==len) t[now].mark=1;
}
}

queue<int > q;
void build_AC()
{
while(!q.empty()) q.pop();
q.push(0);
while(!q.empty())
{
int x=q.front();q.pop();
for(int i=1;i<=10;i++)
{
if(t[x].son[i])
{
t[t[x].son[i]].fail=x?t[t[x].fail].son[i]:0;
q.push(t[x].son[i]);
}
else t[x].son[i]=t[t[x].fail].son[i];
}
if(t[t[x].fail].mark) t[x].mark=1;
}
}

void init()
{
scanf("%s",ss+1);
sl=strlen(ss+1);
scanf("%d",&m);
tot=0;upd(0);
for(int i=1;i<=m;i++) read_trie();
build_AC();
}

int check()
{
for(int i=1;i<=sl;i++)
{
bool p=1;
int now=0;
for(int j=i;j>=1;j--)
{
if(t[ t[now].son[ss[j]-'0'+1] ].mark) {p=0;break;}
now=t[now].son[ss[j]-'0'+1];
}
if(!p) return i;
}
return 0;
}

int f[Maxn][Maxn];
void dp()
{
memset(f,0,sizeof(f));
for(int i=0;i<=tot;i++) f[0][i]=1;//走到i点，继续填0个数的方案

for(int i=1;i<=sl;i++)
{
for(int j=0;j<=tot;j++) if(!t[j].mark)
{
for(int k=1;k<=10;k++) if(!t[t[j].son[k]].mark)
f[i][j]=(f[i][j]+f[i-1][t[j].son[k]])%Mod;
}
}

int ans=0;
if(sl!=1)
{
for(int j=2;j<=sl;j++)
for(int i=2;i<=10;i++) if(!t[t[0].son[i]].mark)
ans=(ans+f[sl-j][t[0].son[i]])%Mod;
}

int now=0;
bool ok=1;
for(int i=sl;i>=1;i--)
{
for(int k=0;k<ss[sl-i+1]-'0';k++)//枚举第i位填的数
{
if(i==sl&&k==0) continue;
if(t[t[now].son[k+1]].mark) continue;
ans=(ans+f[i-1][t[now].son[k+1]])%Mod;
}
now=t[now].son[ss[sl-i+1]-'0'+1];
if(t[now].mark) {ok=0;break;}
}
if(ok) ans=(ans+1)%Mod;
if(sl==1&&ss[1]=='0') ans=0;
printf("%d\n",ans);
}

int main()
{
init();
dp();
return 0;
}