【SDOI2014】数数
Description
计算不大于N且不包含S中的数的个数。
Solution
首先计算一个不大于N的数,而且还有一些限制,很明显就是数位DP。一个字符串在另一个字符串集合中操作,很容易想到AC自动机。
如果要做DP,首先有一个i表示从高位到低位,做到第i位。要在AC自动机中匹配,还要加一个j,表示当前在AC自动机中节点为j。然后还要一个经典,0或1,表示当前数位小于N或等于N。所以就要转移一个DP->f[i,j,0…1],很显然是f[i,j,0]转移到f[i+1,p,0],f[i,j,1]转移到f[i+1,p,0]或f[i+1,p,1],现在就要讨论p是什么。因为现在做到的是节点j,所以之前的已经匹配过了,所以假设第i+1为的数字为k,所以要转移到next[j,k]表示当前在AC自动机的节点为j,要接到下一个数字为k的点为next[j,k]可是是他的儿子或祖先或祖先的儿子。注意在转移的时候,如果它可以转移到AC自动机上的某一点,当时这一点是输入的字符串集合的某一末节点时,就转移或统计答案。
Code
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int mo=1000000007;
queue<int> data;
int i,j,k,l,t,n,m,ans,next[10000][10],num,tot,fail[100000];
int f[1300][1510][2],len;
struct node{
int son[10];
bool biao;
}trie[100000];
char s[10000],st[10000];
void ac_automation(){
int head=0,tail=1,x,i;
data.push(0);
while(!data.empty()){
x=data.front();
data.pop();
fo(i,0,9){
if(trie[x].son[i]){
int y=trie[x].son[i],z=fail[x];
while(z&&!trie[z].son[i]) z=fail[z];
if(x!=0)fail[y]=trie[z].son[i];
if (!trie[y].biao)trie[y].biao=trie[trie[z].son[i]].biao;
data.push(trie[x].son[i]);
}
}
}
}
int main(){
freopen("count.in","r",stdin);
freopen("count.out","w",stdout);
scanf("%s",s+1);
scanf("%d",&n);
fo(j,1,n){
scanf("%s",st+1);
int len=strlen(st+1),x=0;
fo(i,1,len){
if(!trie[x].son[st[i]-'0']){
trie[x].son[st[i]-'0']=++num;
x=num;
}
else x=trie[x].son[st[i]-'0'];
}
trie[x].biao=1;
}
len=strlen(s+1);
ac_automation();
fo(i,0,num){
fo(j,0,9){
int x=i;
while(x&&!trie[x].son[j])x=fail[x];
if(trie[x].son[j])next[i][j]=trie[x].son[j];
}
}
f[1][next[0][s[1]-'0']][1]++;
fo(i,1,s[1]-'0'-1){
f[1][next[0][i]][0]++;
}
fo(i,1,len-1){
fo(k,1,9) f[i+1][next[0][k]][0]++;
fo(j,0,num){
if(trie[j].biao)continue;
if(f[i][j][0]){
fo(k,0,9){
f[i+1][next[j][k]][0]=(f[i+1][next[j][k]][0]+f[i][j][0])%mo;
}
}
if(f[i][j][1]){
fo(k,0,s[i+1]-'0'-1){
f[i+1][next[j][k]][0]=(f[i+1][next[j][k]][0]+f[i][j][1])%mo;
}
f[i+1][next[j][s[i+1]-'0']][1]=(f[i+1][next[j][s[i+1]-'0']][1]+f[i][j][1])%mo;
}
}
}
fo(i,0,num) if(!trie[i].biao)
ans=(ans+f[len][i][0]+f[len][i][1])%mo;
printf("%d\n",ans);
}