【Luogu3041】视频游戏的连击(AC自动机,动态规划)
题解
首先构建出AC自动机
然后在AC自动机上面跑DP
转移很显然从Trie树的节点跳到他的儿子节点
但是要注意一个问题,
在计算的时候,每一个节点加入后能够
造成的贡献
要加上他的子串的贡献
至于DP:
设f[i][j]表示已经使用了i个字母
当前在Trie树的第j个节点上面能够产生的最大贡献
很显然,转移到他的儿子节点上面,同时统计贡献即可
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<algorithm>
using namespace std;
#define INF 1000000
int N,K;
int tot=0,ans;
char s[20];
int f[1100][400];
struct Node
{
int vis[3];
int p;
int fail;
}t[5000];
void Add(char *s)
{
int l=strlen(s);
int now=0;
for(int i=0;i<l;++i)
{
if(!t[now].vis[s[i]-'A'])
t[now].vis[s[i]-'A']=++tot;
now=t[now].vis[s[i]-'A'];
}
t[now].p++;
}
void Build()
{
queue<int> Q;
for(int i=0;i<3;++i)
if(t[0].vis[i])Q.push(t[0].vis[i]);
while(!Q.empty())
{
int u=Q.front();Q.pop();
for(int i=0;i<3;++i)
{
if(t[u].vis[i])
{
t[t[u].vis[i]].fail=t[t[u].fail].vis[i];
Q.push(t[u].vis[i]);
}
else
t[u].vis[i]=t[t[u].fail].vis[i];
}
t[u].p+=t[t[u].fail].p;
}
}
void DP()
{
for(int T=0;T<=K;++T)
for(int i=1;i<=tot;++i)
f[T][i]=-INF;
for(int T=1;T<=K;++T)
for(int i=0;i<=tot;++i)
for(int j=0;j<3;++j)
f[T][t[i].vis[j]]=max(f[T][t[i].vis[j]],f[T-1][i]+t[t[i].vis[j]].p);
for(int i=0;i<=tot;++i)ans=max(ans,f[K][i]);
}
int main()
{
scanf("%d%d",&N,&K);
for(int i=1;i<=N;++i)
{
scanf("%s",s);
Add(s);
}
Build();
DP();
printf("%d\n",ans);
return 0;
}