【AC自动机】[UESTC 554][USACO 2012]Video Game Combos

题目大意：给定有ABC组成的串n个，然后请你生成一个长度为K的串求给定的串在生成串中最多被匹配时的次数

输入

3 7
ABA
CB
ABACB

输出：

4

题目解析：这道题目就是对于每一个节点每次处理下一个（第N+1）个是什么字母，然后每次匹配确定当前字符一定有用（一定存在下一个模式串）然后进行匹配的同时进行DP就行了（保证每个节点到根的串被匹配否则不就浪费长度了么。。）

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <iostream>
using namespace std;
#define rep(i,k) for(int (i)=1;(i)<=(k);(i)++)
#define repz(i,k) for(int (i)=0;(i)<(k);(i)++)
const int MAXN = 20;
const int MAXK = 1000;
struct State{
    int fail, flag;
    int ch[3];
}p[MAXN*15+30];
int scnt, Bef[MAXN*15+30];
void Add(char *s){
    int now = 0, len = strlen(s);
    repz(i, len){
        if(!p[now].ch[s[i]-'A'])
            p[now].ch[s[i]-'A'] = ++scnt;
        now = p[now].ch[s[i]-'A'];
    }
    p[now].flag++;
}
void _initTree(){
    queue<int> que;
    repz(i, 3)
        if(p[0].ch[i])
            que.push(p[0].ch[i]);
    while(!que.empty()){
        int u = que.front();
        que.pop();
        repz(i, 3) if(p[u].ch[i]){
            int v = p[u].ch[i];
            que.push(v);
            int k = p[u].fail;
            while(k && !p[k].ch[i]) k = p[k].fail;
            p[v].fail = p[k].ch[i];
            Bef[v] = p[p[v].fail].flag > 0 ? p[v].fail : Bef[p[v].fail];
        }
    }
}
int dp[MAXK+10][MAXN*15+30];
void match(int len){
    memset(dp, -1, sizeof dp);
    dp[0][0] = 0;
    queue<pair<int, int> > que;
    que.push(make_pair(0, 0));
    pair<int, int> tmp;
    while(!que.empty()){
        tmp = que.front(); que.pop();
        int u = tmp.first;
        int ru = u, i = tmp.second;
        if(i == len) break;
        repz(ids, 3){
            u = ru;
            if(p[u].ch[ids])
                u = p[u].ch[ids];
            else{
                int k = p[u].fail;
                while(k && !p[k].ch[ids]) k = p[k].fail;
                u = p[k].ch[ids];
            }
            if(!u) continue;
            int k = u, sum = 0;
            while(k){
                sum += p[k].flag;
                k = Bef[k];
            }
            if(dp[i][ru] + sum > dp[i+1][u]){
                dp[i+1][u] = dp[i][ru] + sum;
                que.push(make_pair(u, i+1));
            }
        }
    }
    int ans = 0;
    scnt++;
    repz(i, scnt)
        ans = max(ans, dp[len][i]);
    printf("%d\n", ans);
}
int main(){
    int n, k;
    char s[40];
    scanf("%d%d", &n, &k);
    rep(i, n){
        scanf("%s", s);
        Add(s);
    }
    _initTree();
    match(k);

    return 0;
}