洛谷P4052 [JSOI2007]文本生成器 AC自动机+dp

时间:2022-11-23 22:32:47

正解:AC自动机+dp

解题报告:

传送门!

感觉AC自动机套dp的题还挺套路的,,,

一般就先跑遍AC自动机,然后就用dp

dp的状态一般都是f[i][j]:有i个字符,是ac自动机上的第j个节点,然后有的题目可能还要加一维用来满足一些额外要求之类的

然后听说矩阵优化dp挺常见的,,,但我还没做过这种题QAQ

然后这题就直接很套路啊,,,直接上面那种套路一点修改都没有,,,

就显然答案是所有状态-非法状态

所有状态就26m

非法状态就上面那个套路算下

然后就做完了呢,,,

放下代码趴QAQ

#include<bits/stdc++.h>
using namespace std;
#define il inline
#define gc getchar()
#define ll long long
#define ri register int
#define rb register bool
#define rc register char
#define rp(i,x,y) for(ri i=x;i<=y;++i)
#define my(i,x,y) for(ri i=x;i>=y;--i) const int N=+,mod=;
int n,m,nod_cnt,as,f[N][N*];
struct nod{int to[],fail;bool flg;}tr[N*];
char str[N];
queue<int>Q; il int read()
{
rc ch=gc;ri x=;rb y=;
while(ch!='-' && (ch>'' || ch<''))ch=gc;
if(ch=='-')ch=gc,y=;
while(ch>='' && ch<='')x=(x<<)+(x<<)+(ch^''),ch=gc;
return y?x:x;
}
il void insert(char *s)
{
ri lth=strlen(s+),nw=;
rp(i,,lth)
{
if(!tr[nw].to[s[i]-'A'+])tr[nw].to[s[i]-'A'+]=++nod_cnt;
nw=tr[nw].to[s[i]-'A'+];
}
tr[nw].flg=;
}
il void bfs()
{
rp(i,,)if(tr[].to[i])Q.push(tr[].to[i]);
while(!Q.empty())
{
ri nw=Q.front();Q.pop();tr[nw].flg|=tr[tr[nw].fail].flg;
rp(i,,)
if(tr[nw].to[i])Q.push(tr[nw].to[i]),tr[tr[nw].to[i]].fail=tr[tr[nw].fail].to[i];
else tr[nw].to[i]=tr[tr[nw].fail].to[i];
}
}
il int power(ri x,ri y){ri ret=;while(y){if(y&)ret=ret*x%mod;x=x*x%mod;y>>=;}return ret;} int main()
{
n=read();m=read();rp(i,,n){scanf("%s",str+);insert(str);}bfs();
as=power(,m);f[][]=;
rp(i,,m-)
rp(j,,nod_cnt)
if(!tr[j].flg)
rp(k,,)
if(!tr[tr[j].to[k]].flg)f[i+][tr[j].to[k]]=(f[i+][tr[j].to[k]]+f[i][j])%mod;
rp(i,,nod_cnt)as=(as+mod-f[m][i])%mod;
printf("%d\n",as);
return ;
}