bzoj 3530 [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

【分析】

Orz 阿当学长  传送门

【代码】

#include<iostream>#include<cstring>#include<cstdio>#include<queue>#define ll long long#define p 1000000007#define M(a) memset(a,0,sizeof a)#define fo(i,j,k) for(int i=j;i<=k;i++)using namespace std;const int mxn=1505;queue <int> q;char s[mxn];int m,T,num,ans;int n[mxn];int dp[mxn][mxn][2];struct node {int son[10],fail,cnt;} a[mxn];inline void trie(){scanf("%s",s+1);    int x=0,len=strlen(s+1);    fo(i,1,len)    {        int t=s[i]-'0';        if(!a[x].son[t])          a[x].son[t]=(++num);        x=a[x].son[t];    }    a[x].cnt=1;}inline void build(){    fo(i,0,9)      if(a[0].son[i]) q.push(a[0].son[i]);    while(!q.empty())    {        int x=q.front();q.pop();        int fail=a[x].fail;        fo(i,0,9)        {            int y=a[x].son[i];            if(y) a[y].fail=a[fail].son[i],q.push(y);            else a[x].son[i]=a[fail].son[i];        }        a[x].cnt|=a[fail].cnt;    }}inline void dynamic(int len){M(dp);fo(i,1,9)  if(!a[a[0].son[i]].cnt)    dp[1][a[0].son[i]][0]++;fo(i,1,len-2)  fo(j,0,num)    fo(k,0,9) if(!a[a[j].son[k]].cnt)      (dp[i+1][a[j].son[k]][0]+=dp[i][j][0])%=p;fo(i,1,len-1) fo(j,0,num) (ans+=dp[i][j][0])%=p;M(dp);fo(i,1,n[1])  if(!a[a[0].son[i]].cnt)    dp[1][a[0].son[i]][i==n[1]]++;fo(i,1,len-1)  fo(j,0,num)    fo(k,0,9) if(!a[a[j].son[k]].cnt)    {    (dp[i+1][a[j].son[k]][0]+=dp[i][j][0])%=p;    if(k<n[i+1]) (dp[i+1][a[j].son[k]][0]+=dp[i][j][1])%=p;    else if(k==n[i+1]) (dp[i+1][a[j].son[k]][1]+=dp[i][j][1])%=p;}fo(j,0,num) (ans+=dp[len][j][0])%=p,(ans+=dp[len][j][1])%=p;}int main(){scanf("%s",s+1);int len=strlen(s+1);fo(i,1,len) n[i]=s[i]-'0';scanf("%d",&m);fo(i,1,m) trie();build();dynamic(len);printf("%d\n",ans);return 0;}