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

将所有串建成AC自动机。
f[i][j][0/1][0/1] f [ i ] [ j ] [ 0 / 1 ] [ 0 / 1 ] 表示做到第 i i 位,匹配到AC自动机上 j 号点,是否出现过不为 0 0 <script type="math/tex" id="MathJax-Element-362">0</script> 的数,是否卡着上界的方案数,直接DP就好了。
注意判断当前是前导零时,AC自动机上的点要停留在根。

#include<bits/stdc++.h>
using namespace std;
const int N=1510;
const int M=1000000007;
int k,n,m,a[N];
int q[N];
int f[N],nx[N][10],num;
int dp[2][N][2][2];
char s[N];
bool v[N];
vector<int>g[N];
void Insert(int& x,int y,int l){
    if(!x)x=++num;else if(v[x])return;
    if(y==l){
        v[x]=1;
        return;
    }
    Insert(nx[x][s[y]-'0'],y+1,l);
}
void Dfs(int x){
    for(int i=0;i<g[x].size();i++){
        int t=g[x][i];
        v[t]|=v[x];
        Dfs(t);
    }
}
inline void Build(){
    int l=0,r=0;
    for(int i=0;i<10;i++)
    if(nx[0][i])q[++r]=nx[0][i];
    while(++l<=r){
        int x=q[l];
        for(int i=0;i<10;i++)
        if(nx[x][i]){
            f[nx[x][i]]=nx[f[x]][i];
            q[++r]=nx[x][i];
        }else nx[x][i]=nx[f[x]][i];
    }
    for(int i=1;i<=num;i++)g[f[i]].push_back(i);
    Dfs(0);
}
inline void Add(int& x,int y){
    x=(x+y)%M;
}
inline int Solve(){
    dp[1][0][1][0]=1;
    bool b=0;
    for(int i=1;i<=n;i++,b^=1){
        memset(dp[b],0,sizeof(dp[b]));
        for(int j=0;j<=num;j++)
        for(int k=0;k<2;k++)
        for(int l=0;l<2;l++)
        if(dp[b^1][j][k][l]){
            int mx=k?a[i]:9;
            for(int p=0;p<=mx;p++){
                int K=(k&&p==mx);
                if(!l&&!p){
                    Add(dp[b][0][K][0],dp[b^1][j][k][l]);
                    continue;
                }
                if(!v[nx[j][p]])Add(dp[b][nx[j][p]][K][1],dp[b^1][j][k][l]);
            }
        }
    }
    int Ans=0;
    for(int i=0;i<=num;i++)
    for(int k=0;k<2;k++)
    Add(Ans,dp[b^1][i][k][1]);
    return Ans;
}
int main(){
    scanf("%s",s);
    n=strlen(s);
    for(int i=1;i<=n;i++)a[i]=s[i-1]-'0';
    scanf("%d",&m);
    while(m--){
        scanf("%s",s);
        Insert(nx[0][s[0]-'0'],1,strlen(s));
    }
    Build();
    cout<<Solve()<<endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值