「SDOI2014」数数

「SDOI2014」数数

谨以此题纪念第一次做AC自动机类问题。

大意

求比n小的能够满足不含有任意一个串的数字个数。

思路

看到比n小，可以很容易的想到要数位dp，看到和多个串匹配，可以想到要用AC自动机。
那么正解就是 在AC自动机上做数位dp。
状态是$f[i][j]$表示第i位和AC自动机上的第j个节点匹配的方案数(然而我并没有见过这么做的,果断的gg了)。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
const int maxn=2000+5;
const ll Mod=1000000000+7;
char s[maxn],s1[maxn];
struct node
{
    int son[11],fail; bool mark;
}ac[maxn*100];
int m,tot,q[maxn*100],len1;
ll ans,f[2000][2505];
void insert(char *ch)
{
    int len=strlen(ch),root=0;
    for(int i=0; i<len; i++)
    {
        int &d=ac[root].son[ch[i]-'0'];
        if(!d) d=++tot;
        root=d;
    }
    ac[root].mark=true;
}
void getfail()//¹ãËѽ¨Á¢failÖ¸Õë 
{
    int head=0,tail=0;
    for(int i=0; i<=9; i++)
    {
        int d=ac[0].son[i];
        if(d)
        {
            ac[d].fail=0; tail++; q[tail]=d;
        }
    }
    while(head<tail)
    {
        head++; int now=q[head];
        for(int i=0; i<=9; i++)
        {
            int d=ac[now].son[i];
            if(!d)
            {
                ac[now].son[i]=ac[ac[now].fail].son[i]; continue;
            }
            else
            {
                ac[d].fail=ac[ac[now].fail].son[i]; tail++; q[tail]=d;
            }
        }
    }
}
ll dfs(int pos,int x,bool limit)
{
    if(ac[x].mark) return 0;//Èç¹ûµ±Ç°µÄµãÊÇij¸ö´®µÄÖյ㣬¼´Æ¥ÅäÉÏÁË¡£ 
    if(pos==len1-1) return 1;//×ßÍêÒ»¸öÊý×ÖÁË,ÊýλdpµÄ±ê×¼¡£ 
    if(!limit && f[pos][x]!=-1) return f[pos][x];//¼ÇÒ仯£¬µ«Ö»¼ÇûÉϽçµÄ¡£ 
    ll ret=0; int up;
    up=limit?s[pos+1]-'0':9;//ö¾ÙÖյ㡣 µ±Ç°Î»ÒѾ­Ñ¡ºÃÁË£¬ÕÒÏÂһλµÄ 
    for(int i=0; i<=up; i++)
    {
        ret+=dfs(pos+1,ac[x].son[i],limit && i==up); ret%=Mod;
    }
    if(!limit) f[pos][x]=ret;//¼ÇÒ仯 
    return ret;
}
int main()
{
    freopen("test.in","r",stdin);
    freopen("test.out","w",stdout);
    scanf("%s",s);
    scanf("%d",&m);
    for(int i=1; i<=m; i++)
    {
        scanf("%s",s1);
        insert(s1);
    }
    getfail();
    memset(f,-1,sizeof(f));
    len1=strlen(s);
    for(int i=0; i<len1; i++)//ÕâôÏÈö¾Ù¿ÉÒÔÈ¥³ýÇ°µ¼0µÄÓ°Ïì 
    {
        for(int j=1; j<=(i==0?s[0]-'0':9); j++)
        {
            ans+=dfs(i,ac[0].son[j],i==0 && j==s[0]-'0'); ans%=Mod;
//          cout<<ans<<endl;
        }
    }
    printf("%lld\n",ans);
    return 0;
}