bzoj3530【SDOI2014】数数

3530: [Sdoi2014]数数

Time Limit: 10 Sec   Memory Limit: 512 MB
Submit: 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

Sample Output

14

HINT

 下表中l表示N的长度，L表示S中所有串长度之和。

1 < =l < =1200 , 1 < =M < =100 ,1 < =L < =1500

Source

Round 1 day 1

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);
}