bzoj 1030 AC自动机+dp

时间:2024-06-24 22:35:08

bzoj 1030 AC自动机+dp

代码:

//先把给的单词建AC自动机并且转移fail,然后d[i][j]表示构造的文章到第i位时处在字典树的第j个节点的不包含单词的数量,最后用总的数量26^m
//-d[m][0~sz]即可。其中不能走单词结尾的节点以及他们的fail。这里其实要把每个节点都连向他的26个后继,但是不连也没关系可以看作
//那些没出现的节点都转移成了0节点。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int MAXN=;
const int MOD=;
int node[MAXN][],val[MAXN],f[MAXN],sz,d[][MAXN];
char s[];
int n,m;
void init()
{
sz=;
memset(node[],,sizeof(node[]));
val[]=f[]=;
memset(d,,sizeof(d));
}
void insert(char *s)
{
int len=strlen(s),rt=;
for(int i=;i<len;i++){
int id=s[i]-'A';
if(!node[rt][id]){
node[rt][id]=++sz;
memset(node[sz],,sizeof(node[sz]));
val[sz]=;
}
rt=node[rt][id];
}
val[rt]=;
}
void get_fail()
{
queue<int>q;
for(int i=;i<;i++){
int u=node[][i];
if(u) { q.push(u);f[u]=; }
}
while(!q.empty()){
int rt=q.front();q.pop();
for(int i=;i<;i++){
int u=node[rt][i];
if(!u){
node[rt][i]=node[f[rt]][i];
continue;
}
q.push(u);
f[u]=node[f[rt]][i];
}
val[rt]|=val[f[rt]];
}
}
void get_d()
{
d[][]=;
for(int i=;i<=m;i++){
for(int j=;j<=sz;j++){
if(val[j]||d[i-][j]==) continue;
for(int k=;k<;k++)if(!val[node[j][k]])
d[i][node[j][k]]=(d[i][node[j][k]]+d[i-][j])%MOD;
}
}
}
int main()
{
init();
scanf("%d%d",&n,&m);
for(int i=;i<n;i++){
scanf("%s",s);
insert(s);
}
get_fail();
get_d();
int ans1=,ans2=;
for(int i=;i<=sz;i++)if(!val[i])
ans1=(ans1+d[m][i])%MOD;
for(int i=;i<=m;i++)
ans2=(ans2*)%MOD;
printf("%d\n",(ans2+MOD-ans1)%MOD);
return ;
}