BZOJ 1030 [JSOI2007] 文本生成器 AC自动机+DP

题目大意:给出n个模式串,给定m,表示将会随机生成长度为m的串,求出随机生成的串中,包含模式串的串的个数%10007的值

若直接求包含模式串的串的个数,DP状态不好转移(可能在串的任意位置出现模式串,可能出现多个模式串,不方便统计方案数)。所以可以利用补集的思想,用总数减去不包含的串的个数即为答案。

这样就方便设计状态了。设状态f(i,j)表示在自动机中的i号结点匹配了长度为j的串仍为出现模式串的方案数。当第i号结点的儿子k不包含模式串时,f(k,j+1)+=f(i,j)。最后统计一下每个结点上匹配了m个字符后仍未出现模式串的方案个数即可。

#include <cstdio>
#include <cstring>
#include <queue>
#define N 50005
#define MOD 10007
using namespace std;
struct Node {
    Node *ch[26],*nxt;
    bool ext;
    Node():nxt(0x0),ext(false) {memset(ch,0,sizeof ch);}
}p[N],*root=&p[0];
char s[N];
int n,m,top;
int f[N][105];
void Insert() {
    int len=strlen(s);
    Node* o=root;
    for(int i=0;i<len;i++) {
        int z=s[i]-'A';
        if(!o->ch[z]) o->ch[z]=&p[++top];
        o=o->ch[z];
    }
    o->ext=true;
    return ;
}
void getFail() {
    static queue<Node*> q;
    Node* o=root;
    for(int i=0;i<26;i++)
        if(o->ch[i]) q.push(o->ch[i]) , o->ch[i]->nxt=o;
        else o->ch[i]=o;
    while(!q.empty()) {
        o=q.front(); q.pop();
        for(int i=0;i<26;i++)
            if(o->ch[i]) q.push(o->ch[i]) , o->ch[i]->nxt=o->nxt->ch[i];
            else o->ch[i]=o->nxt->ch[i];
        Node* tmp=o->nxt;
        while(tmp!=root && !tmp->ext) tmp=tmp->nxt;
        o->nxt=tmp;
        if(tmp->ext) o->ext=true;
    }
    return ;
}
void Match() {
    f[0][0]=1;
    for(int k=0;k<m;k++)
        for(int i=0;i<=top;i++) {
            if(p[i].ext) continue;
            for(int j=0;j<26;j++)
                if(!p[i].ch[j]->ext)
                    f[p[i].ch[j]-p][k+1]=(f[p[i].ch[j]-p][k+1]+f[i][k])%MOD;
        }
    return ;
}
int f_pow(int x,int y) {
    int cnt=1;
    while(y) {
        if(y&1) cnt=cnt*x%MOD;
        x=x*x%MOD;
        y>>=1;
    }
    return cnt;
}
int main() {
    scanf("%d%d",&n,&m);
    while(n--) scanf("%s",s) , Insert();
    getFail();
    Match();
    int ans=0;
    for(int i=0;i<=top;i++) ans=(ans+f[i][m])%MOD;
    int sum=f_pow(26,m);
    printf("%d\n",(sum-ans+MOD)%MOD);
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值