bzoj 3530: [Sdoi2014]数数

最新推荐文章于 2021-01-22 16:24:13 发布

原创最新推荐文章于 2021-01-22 16:24:13 发布 · 261 阅读

0 ·

CC 4.0 BY-SA版权

AC自动机同时被 2 个专栏收录

6 篇文章

订阅专栏

数位dp

4 篇文章

订阅专栏

本文介绍了一种结合数位DP与自动机的方法来解决一类特定字符串匹配问题。通过构造自动机并采用数位DP进行状态转移，该方法能够高效地计算1到n范围内不含特定子串的数的数量。

题意：

求1~n中不含有集合S中的元素为子串，不含前导0。

题解：

~~一开始傻傻的写了之前的数位dp版本。~~
换了种打法， $f[i][j][k]$ 表示前i位，在自动机j号点上。
k=0为小于n，k=1等于，k=2大于。
然后瞎转移。
code：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<queue>
#include<algorithm>
#define LL long long
using namespace std;
const int mod=1000000007;
struct trnode{
    int a[11],fail;
    bool tail;
    trnode(){tail=false;}
}tr[2000];int root=1,tot=1;
int f[1210][2000][4];
int m,n,a[1210];
char s[1510];
void build()
{
    int len=strlen(s+1);
    int x=root;
    for(int i=1;i<=len;i++)
    {
        int c=s[i]-'0';
        if(tr[x].a[c]) x=tr[x].a[c];
        else x=tr[x].a[c]=++tot;
    }
    tr[x].tail=true;
}
queue<int> q;
void make_root()
{
    q.push(root);
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(int i=0;i<=9;i++)
        {
            if(!tr[x].a[i]) continue;
            if(x==root) tr[tr[x].a[i]].fail=root;
            else
            {
                int j=tr[x].fail;
                while(j!=root&&!tr[j].a[i]) j=tr[j].fail;
                tr[tr[x].a[i]].fail=tr[j].a[i];
                tr[tr[x].a[i]].tail|=tr[tr[j].a[i]].tail;
            }
            q.push(tr[x].a[i]);
        }
    }
}
int qs(int k,int x)
{
    if(k==x) return 1;
    if(k<x) return 0;
    return 2;
}
int main()
{
    scanf("%s",s+1);
    n=strlen(s+1);
    for(int i=1;i<=n;i++) a[i]=s[i]-'0';
    scanf("%d",&m);
    for(int i=0;i<=9;i++) tr[root].a[i]=++tot;
    for(int i=1;i<=m;i++)
    {
        scanf("%s",s+1);
        build();
    }
    make_root();
    f[0][root][1]=1;
    for(int i=0;i<n;i++)
        for(int x=1;x<=tot;x++)
            for(int K=0;K<3;K++)
            {
                if(tr[x].tail||f[i][x][K]==0) continue;
                for(int c=0;c<=9;c++)
                {
                    if(x==root&&c==0) continue;
                    int k=x;
                    while(!tr[k].a[c]) k=tr[k].fail;
                    if(!tr[tr[k].a[c]].tail)
                    {
                        if(K!=1) (f[i+1][tr[k].a[c]][K]+=f[i][x][K])%=mod;
                        else (f[i+1][tr[k].a[c]][qs(c,a[i+1])]+=f[i][x][K])%=mod;
                    }
                }
            }
    int ans=0;
    for(int i=1;i<=n;i++)
        for(int j=2;j<=tot;j++)
            for(int k=0;k<=2;k++)
                if(k<=1||i!=n) (ans+=f[i][j][k])%=mod;
    printf("%d",ans);
}