【SDOI 2014】数数

传送门

---

Problem

我们称一个正整数 $n$ 是幸运数，当且仅当它的十进制表示中不包含数字串集合 $S$ 中任意一个元素作为其子串。

例如当 $S=(22,333,0233)$ 时，$233$ 是幸运数，$2333$、$20233$、$3223$ 不是幸运数。

给定 $n$ 和 $S$，计算不大于 $n$ 的幸运数个数。

---

Solution

比较妙的一道题。

这道题其实比较像 [POI 2000] 病毒，也是要让 $S$ 中任意元素都不是 $n$ 的子串。

于是我们按照 $S$ 建立AC自动机，在自动机上跑的时候就不能到单词尾节点。

又由于这里的 $n$ 表示 $1$ ~ $n$ 的一个范围，于是想到数位DP。

那这道题就是在AC自动机上跑数位DP统计数量，注意到不要跑到尾节点就可以了。

注意前导零。

---

Code

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define Mod 1000000007
using namespace std;
int tot=0,len;
struct Trie{
	int son[10];
	int fail,end;
}a[1505];
char N[1205],S[1505];
int f[1205][1505];
void insert(){
	int i,p=0;
	int l=strlen(S);
	for(i=0;i<l;++i){
		int x=S[i]-'0';
		if(!a[p].son[x])
			a[p].son[x]=++tot;
		p=a[p].son[x];
	}
	a[p].end=1;
}
void Get_fail(){
	int x,i;
	queue<int>Q;
	for(i=0;i<10;++i)
		if(a[0].son[i])
			Q.push(a[0].son[i]);
	while(!Q.empty()){
		x=Q.front();Q.pop();
		for(i=0;i<10;++i){
			if(a[x].son[i]){
				Q.push(a[x].son[i]);
				a[a[x].son[i]].fail=a[a[x].fail].son[i];
				if(a[a[x].fail].end)  a[x].end=1;
			}
			else  a[x].son[i]=a[a[x].fail].son[i];
		}
	}
}
int dp(int pos,int x,bool limit,bool lead){
	if(pos>len)  return !lead;
	if(!limit&&!lead&&~f[pos][x])  return f[pos][x];
	int i,up=limit?(N[pos]-'0'):9,ans=0;
	for(i=0;i<=up;++i){
		if(!a[a[lead?0:x].son[i]].end){
			if(lead)  ans=(ans+dp(pos+1,a[0].son[i],limit&&i==up,lead&&i==0))%Mod;
			else  ans=(ans+dp(pos+1,a[x].son[i],limit&&i==up,false))%Mod;
		}
	}
	if(!limit&&!lead)  f[pos][x]=ans;
	return ans;
}
int main(){
	int n,i;
	scanf("%s%d",N+1,&n);
	len=strlen(N+1);
	for(i=1;i<=n;++i){
		scanf("%s",S),insert();
	}
	Get_fail();
	memset(f,-1,sizeof(f));
	printf("%d",dp(1,0,true,true));
	return 0;
}