3530: [Sdoi2014]数数
Time Limit: 10 Sec Memory Limit: 512 MBSubmit: 530 Solved: 298
[ Submit][ Status][ Discuss]
Description
我们称一个正整数N是幸运数,当且仅当它的十进制表示中不包含数字串集合S中任意一个元素作为其子串。例如当S=(22,333,0233)时,233是幸运数,2333、20233、3223不是幸运数。
给定N和S,计算不大于N的幸运数个数。
Input
输入的第一行包含整数N。
接下来一行一个整数M,表示S中元素的数量。
接下来M行,每行一个数字串,表示S中的一个元素。
Output
输出一行一个整数,表示答案模109+7的值。
Sample Input
20
3
2
3
14
3
2
3
14
Sample Output
14
HINT
下表中l表示N的长度,L表示S中所有串长度之和。
1 < =l < =1200 , 1 < =M < =100 ,1 < =L < =1500
Source
AC自动机+动态规划,思路很棒
首先在Trie树中加入失配边,形成一个新的图。
然后分两种情况进行动态规划,分别为位数小于l和等于l。当位数小于l时,没有什么限制,直接转移就可以了;当位数等于l时,需要多加一维表示前i位的数是否等于n的前i位。(具体转移详见代码)
另外这道题还有一点需要注意:因为幸运数是没有前导零的,所以在图中第一步不能走t[1][0]。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<queue>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define pa pair<int,int>
#define maxn 1510
#define mod 1000000007
using namespace std;
int t[maxn][10],go[maxn],f[1210][maxn][2],a[maxn];
int tot=1,n,l,ans=0;
char s[maxn];
bool v[maxn];
queue<int> q;
inline void insert()
{
scanf("%s",s);
int len=strlen(s),now=1;
F(i,0,len-1)
{
int x=s[i]-'0';
if (!t[now][x]) t[now][x]=++tot;
now=t[now][x];
}
v[now]=1;
}
inline void bfs()
{
q.push(1);
while (!q.empty())
{
int x=q.front(),y,j;q.pop();v[x]|=v[go[x]];
F(i,0,9)
{
j=go[x];
while (j&&!t[j][i]) j=go[j];
if (t[x][i])
{
go[y=t[x][i]]=j?t[j][i]:1;
q.push(y);
}
else t[x][i]=j?t[j][i]:1;
}
}
}
int main()
{
scanf("%s",s);
l=strlen(s);
F(i,0,l-1) a[i]=s[i]-'0';
scanf("%d",&n);
F(i,1,n) insert();
bfs();
memset(f,0,sizeof(f));
F(i,1,9) if (!v[t[1][i]]) f[1][t[1][i]][0]+=1;
F(i,1,l-2) F(j,1,tot) F(x,0,9) if (!v[t[j][x]])
(f[i+1][t[j][x]][0]+=f[i][j][0])%=mod;
F(i,1,l-1) F(j,1,tot) (ans+=f[i][j][0])%=mod;
memset(f,0,sizeof(f));
F(i,1,a[0]) if (!v[t[1][i]]) f[1][t[1][i]][i==a[0]]+=1;
F(i,1,l-1) F(j,1,tot) F(x,0,9) if (!v[t[j][x]])
{
(f[i+1][t[j][x]][0]+=f[i][j][0])%=mod;
if (x<a[i]) (f[i+1][t[j][x]][0]+=f[i][j][1])%=mod;
else if (x==a[i]) (f[i+1][t[j][x]][1]+=f[i][j][1])%=mod;
}
F(i,1,tot)
{
(ans+=f[l][i][0])%=mod;
(ans+=f[l][i][1])%=mod;
}
printf("%d\n",ans);
}