bzoj3530 [Sdoi2014]数数
原题地址:http://www.lydsy.com/JudgeOnline/problem.php?id=3530
题意:
我们称一个正整数N是幸运数,当且仅当它的十进制表示中不包含数字串集合S中任意一个元素作为其子串。例如当S=(22,333,0233)时,233是幸运数,2333、20233、3223不是幸运数。
给定N和S,计算不大于N的幸运数个数。
数据范围
下表中l表示N的长度,L表示S中所有串长度之和。
1 < =l < =1200 , 1 < =M < =100 ,1 < =L < =1500
题解:
要求”不大于”,想到数位DP。
然后这个”不包含子串”需要AC自动机。
用的自己一般的数位dp打法:
dp[i][j][1/0][1/0]表示,当前第几位,AC自动机上何处,有没有lim,有没有前导零。
注意:
给出的串可能有前导零,所以当目前都是前导零时,不能走到ch[i][0]而只能留在根
代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int N=1510;
const int mod=1000000007;
queue<int> Q;
int n=0,m,ch[N][10],fail[N],isw[N],root=0,tail=0,num[N],dp[N][N][2][2];
char s[N];
void insert()
{
int len=strlen(s); int tmp=0;
for(int i=0;i<len;i++)
{
int c=s[i]-'0';
if(!ch[tmp][c]) ch[tmp][c]=++tail;
tmp=ch[tmp][c];
}
isw[tmp]=1;
}
void getfail()
{
for(int i=0;i<10;i++) if(ch[root][i]) Q.push(ch[root][i]);
while(!Q.empty())
{
int tmp=Q.front(); Q.pop();
for(int i=0;i<10;i++)
{
if(!ch[tmp][i]) {ch[tmp][i]=ch[fail[tmp]][i];continue;}
int u=ch[tmp][i];
fail[u]=ch[fail[tmp]][i];
isw[u]|=isw[fail[u]];
Q.push(u);
}
}
}
int main()
{
scanf("%s%d",s,&m);
n=strlen(s); for(int i=0;i<n;i++) num[i]=s[i]-'0';
for(int i=1;i<=m;i++) {scanf("%s",s); insert();}
getfail();
dp[0][root][1][1]=1;
for(int i=0;i<n;i++)
for(int j=0;j<=tail;j++)
for(int lim=0;lim<=1;lim++)
for(int z=0;z<=1;z++)
if(dp[i][j][lim][z])
{
int top=lim?num[i]:9;
for(int c=0;c<=top;c++)
{
if(isw[ch[j][c]]) continue;
int nlim=(lim&&c==top); int nz=(z&&c==0);
if(!nz) dp[i+1][ch[j][c]][nlim][nz]=(dp[i+1][ch[j][c]][nlim][nz]+dp[i][j][lim][z])%mod;
else dp[i+1][root][nlim][nz]=(dp[i+1][root][nlim][nz]+dp[i][j][lim][z])%mod;
}
}
int ret=0;
for(int i=0;i<=tail;i++)
for(int lim=0;lim<=1;lim++) ret=(ret+dp[n][i][lim][0])%mod;
printf("%d\n",ret);
return 0;
}