P3041AC自动机+dp

第一道ac自动机加dp的题目
1a了很开心
据说ac自动机的dp都是有套路的，就是状态表示大都是2维，dp[i][j],表示串匹配了i个，最后一位停留在了trie图的j号点上了
这题是模板题，
以下给出ac代码

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>

using namespace std;
const int maxn=1010,maxm=25,inf=1e8;
int trie[maxn][3],cnt,fail[maxn],val[maxn];
int n,m;
char s[maxm];
int dp[maxn][400];

void add(){
    int len=strlen(s),now=0,id;
    for(int i=0;i<len;i++){
        id=s[i]-'A';
        if(!trie[now][id])trie[now][id]=++cnt;
        now=trie[now][id];
    }
    val[now]++;
}
void build(){
    queue<int>q;
    for(int i=0;i<3;i++)if(trie[0][i])q.push(trie[0][i]);
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=0;i<3;i++)
            if(trie[x][i])fail[trie[x][i]]=trie[fail[x]][i],q.push(trie[x][i]);
            else trie[x][i]=trie[fail[x]][i];
        val[x]+=val[fail[x]];
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    while(n--)scanf("%s",&s),add();build();
    for(int i=0;i<=m;i++)
        for(int j=1;j<=cnt;j++)dp[i][j]=-inf;
    for(int i=1;i<=m;i++)
        for(int j=0;j<=cnt;j++)
            for(int k=0;k<3;k++)if(trie[j][k])dp[i][trie[j][k]]=max(dp[i][trie[j][k]],dp[i-1][j]+val[trie[j][k]]);
    int ans=0;
    for(int i=0;i<=cnt;i++)ans=max(ans,dp[m][i]);
    printf("%d\n",ans);
    return 0;
}