[BZOJ3530][SDOI2014]数数 AC自动机+数位DP

考虑先建出集合S的AC自动机,并把它补成Trie图(是叫这个吗?),并在AC自动机标记上S集合中每个数字的结束位置。把这个标记沿fail树下传之后,一个串只要在AC自动机上没有走到标记点,就说明它是幸运数.
所以考虑数位DP,设 f[i][j][0/1] 表示从高往低填到第 i 位,跑到AC自动机上 j 点,前 i <N 还是 =N 。转移即可。
考虑到不能有前导零,所以填每一位的时候可以把根节点单独转移一次1~9。
代码:

#include
#include
#include
#define up(y,x) x=((x)+(y))%mod
using namespace std;
const int mod=1000000007;
char lim[1210],cs[1510];
int trie[1510][10],fa[1510],dl[1510],n,m,tot,f[1210][1510][2];
bool pl[1510];
void ins()
{
    int p=0;
    for(int l=strlen(cs+1),i=1;i<=l;i++)
    {
        if(!trie[p][cs[i]-'0']) trie[p][cs[i]-'0']=++tot;
        p=trie[p][cs[i]-'0'];
    }
    pl[p]=1;
}
void getac()
{
    for(int head=1,tail=1,v=0;head<=tail;v=dl[++head])
        for(int c=0;c<=9;c++)
            if(trie[v][c])
            {
                int p=fa[v];
                for(;p&&!trie[p][c];p=fa[p]);
                if(trie[p][c]!=trie[v][c]) fa[trie[v][c]]=trie[p][c];
                pl[trie[v][c]]|=pl[trie[p][c]];
                dl[++tail]=trie[v][c];
            }
            else trie[v][c]=trie[fa[v]][c];
}
void solve()
{
    for(int i=1;i<=n;i++)
    {
        for(int c=1,v=trie[0][c];c<=9;v=trie[0][++c])
        {
            if(i!=1||c1,f[i][v][0]);
            else if(c==lim[i]) up(1,f[i][v][1]);
        }
        for(int j=0;j<=tot;j++)
            if(!pl[j])
                for(int c=0,v=trie[j][c];c<=9;v=trie[j][++c])
                {
                    up(f[i-1][j][0],f[i][v][0]);
                    if(c1][j][1],f[i][v][0]);
                    else if(c==lim[i]) up(f[i-1][j][1],f[i][v][1]);
                }
    }
    int ans=0;
    for(int j=0;j<=tot;j++)
        if(!pl[j])  up(f[n][j][0],ans),up(f[n][j][1],ans);  
    printf("%d",ans);   
}
int main()
{
    scanf("%s",lim+1);
    n=strlen(lim+1);
    for(int i=1;i<=n;i++)
        lim[i]-='0';    
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%s",cs+1);
        ins();
    }
    getac();
    solve();
    return 0;
}

你可能感兴趣的:(dp,AC自动机)