bzoj-1030: [JSOI2007]文本生成器(ac自动机+dp)

好像是入门题的说

跑生成的串中不带给定字符串的个数ans   26^m-ans就是答案

dp[i][j]表示字符第i位在ac自动机上第j的节点的合法情况数

如果转移到的下一个节点是字符的结尾,就是非法的

转移dp[i][j]+=dp[i-1][j];

ps 节点编号从1开始

#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
#include<set>
#include<map>
#include<vector>

#define inf 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
#define ll long long
#define sd(x) scanf("%d",&(x))
#define sl(x) scanf("%lld",&(x))
#define scs(s) scanf("%s",s)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
#define lowbit(x) x&(-x)

using namespace std;

const int maxn=1e5+10;
const int mod=10007;

int dp[105][6005];
int mub=1;
int tree[maxn][26],cnt[maxn];
int fail[maxn],End[maxn][26];
char s[maxn];

void Insert()
{
    int root=1;
    int l=strlen(s);
    rep(i,0,l-1)
    {
        int now=s[i]-'A';
        if(!tree[root][now])
            tree[root][now]=++mub;
        root=tree[root][now];
    }
    cnt[root]++;
}

void getfail()
{
    for(int i=1;i<=mub;i++)
        fail[i]=1;
    queue<int>q;fail[1]=0;
    q.push(1);
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        rep(i,0,25)
        {
            int k=fail[now];
            if(tree[now][i])
            {
                q.push(tree[now][i]);
                fail[tree[now][i]]=k?tree[fail[now]][i]:1;
                cnt[tree[now][i]]|=cnt[fail[tree[now][i]]];
            }
            else
                tree[now][i]=k?tree[fail[now]][i]:1;
        }
    }
}

void DP(int x)
{
    rep(i,1,mub)
    {
        rep(j,0,25)
        {
            if(cnt[tree[i][j]]) continue;
            dp[x][tree[i][j]]+=dp[x-1][i];
            dp[x][tree[i][j]]%=mod;
        }
    }
}

int main(){
    int n,m,ans=1,tot=0;
    sd(n),sd(m);
    rep(i,1,n) scs(s),Insert();
    getfail();dp[0][1]=1;
    rep(i,1,m) DP(i),ans=(ans*26)%mod;
    rep(i,1,mub)
        tot=(tot+dp[m][i])%mod;
    printf("%d\n",((ans-tot)%mod+mod)%mod);
    return 0;
}

 

posted @ 2019-08-03 10:27  Minun  阅读(149)  评论(0编辑  收藏