bzoj3530: [Sdoi2014]数数

链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=3530

题解

　　一个小错查了一下午，建trie图的时候我把0当做串的终止，可题目中明明要求0是合法的。。。
　　唉。。
　　f[i][j]表示长度为i的数字匹配到j点的方案数(允许前导零没有大小限制)，g[i][j]是有大小限制的。
　　然后直接dp就行了（ac自动机上的dp道路+数位dp套路）。
　　答案的统计：这个有点费脑子，枚举位数，枚举下一位从1~9，然后计入ans就行了。（其实不费脑子但是我今天有点智障）

代码

//AC自动机上的数位dp#include <cstdio>#include <algorithm>#include <queue>#include <cstring>#define maxn 2000#define mod 1000000007using namespace std;int num[maxn], trie[maxn][10], tail[maxn], tot=1, fail[maxn], N, f[maxn][maxn],    g[maxn][maxn], h[maxn][maxn];queue<int> q;void insert(int *s){    int p;    for(p=1;*s!=-1 and !tail[p];s++)p=trie[p][*s]?trie[p][*s]:trie[p][*s]=++tot;    tail[p]=1;}int acamove(int pos, int x){    for(;pos and !trie[pos][x];pos=fail[pos]);    return pos?trie[pos][x]:1;}void acabuild(){    int u, v, i;    q.push(1);    while(!q.empty())    {        u=q.front();q.pop();        for(i=0;i<=9;i++)        {            if(trie[u][i])            {                v=acamove(fail[u],i);                fail[trie[u][i]]=v;                q.push(trie[u][i]);            }            else trie[u][i]=acamove(u,i);        }    }}void input(){    char s[maxn];    int t[maxn];    int M, i, j, l;    scanf("%s%d",s,&M);    N=strlen(s);    for(i=0;i<N;i++)num[N-i]=s[i]-48;    for(i=1;i<=M;i++)    {        scanf("%s",s);        l=strlen(s);        for(j=0;s[j];j++)t[j]=s[l-j-1]-48;t[l]=-1;        insert(t);    }}int dp(){    int i, j, now, ans=0;    f[0][1]=g[0][1]=1;    for(i=0;i<N;i++)        for(j=1;j<=tot;j++)        {            if(tail[j] or (!f[i][j] and !g[i][j]))continue;            for(now=0;now<=9;now++)                f[i+1][trie[j][now]]=(f[i+1][trie[j][now]]+f[i][j])%mod;            for(now=0;now<=num[i+1];now++)            {                if(now!=num[i+1])                    g[i+1][trie[j][now]]=(g[i+1][trie[j][now]]+f[i][j])%mod;                else g[i+1][trie[j][now]]=(g[i+1][trie[j][now]]+g[i][j])%mod;            }            if(i<N-1)            {                for(now=1;now<=9;now++)                    if(!tail[trie[j][now]])ans=(ans+f[i][j])%mod;            }            else            {                for(now=1;now<=num[N];now++)                {                    if(tail[trie[j][now]])continue;                    if(now==num[N])ans=(ans+g[i][j])%mod;                    else ans=(ans+f[i][j])%mod;                }            }        }    return ans;}int main(){    input();    acabuild();    printf("%d",dp());    return 0;}