AC自动机-模板

#include <cstdio>
#include <queue>
#include <cstring>

using namespace std;

const int _N = 1000000;

int rt = 1, TrieCnt = 1;
queue<int> Q;

struct node {
    int w, fail, son[3];
} T[_N];

char str[300];
int f[1200][3200], N, K;

void Extend()
{
    int p = rt, len = strlen(str+1);
    for (int i = 1; i <= len; ++i) {
        int &son = T[p].son[str[i]-'A'];
        if (!son) son = ++TrieCnt;
        p = son;
    }
    ++T[p].w;
}

void Init()
{
    while (!Q.empty()) Q.pop();
    Q.push(rt), T[rt].fail = rt;
    while (!Q.empty()) {
        int p = Q.front(); Q.pop();
        T[p].w += T[T[p].fail].w;
        for (int i = 0; i < 3; ++i) {
            int &son = T[p].son[i], j;
            for (j = T[p].fail; j != rt && !T[j].son[i]; j = T[j].fail);
            int tmp = T[j].son[i];
            if (tmp && tmp != son) j = tmp;//把 j 改为 son 的父亲的 fail 链上最近的点的儿子（由定义不能为 son 自己） 
            if (son) T[son].fail = j, Q.push(son);
            else son = j;//强行给自己找一个儿子！ 注意这里非常重要！注意这里非常重要！注意这里非常重要！
        }
    }
    return;
}

int DP()
{
    int i, j;
    for (i = 1; i <= TrieCnt; ++i) f[0][i] = -1;
    f[0][rt] = 0;
    for (i = 1; i <= K; ++i) {
        for (j = 1; j <= TrieCnt; ++j)
            f[i][j] = -1;
        for (j = 1; j <= TrieCnt; ++j) {
            if (f[i-1][j] == -1) continue;
            int p = j;
            for (int g = 0; g < 3; ++g) {
                int son = T[p].son[g];
                if (son) f[i][son] = max(f[i][son], f[i-1][j]+T[son].w);
            }
        }
    }
    int ans = 0;
    for (i = 1; i <= TrieCnt; ++i) ans = max(ans, f[K][i]);
    return ans;
}

int main()
{
    scanf("%d%d", &N, &K);
    while (N--) scanf("%s", str+1), Extend();
    Init();
    printf("%d
", DP());
    return 0;
}

NKOJ1960

P1960【自动机】关键词搜索

时间限制 : 10000 MS   空间限制 : 524288 KB

问题描述

如今，搜索引擎像Google,Baidu等搜索引擎已经成为了人们生活的一部分。
韦斯利也想把这个搜索功能加入到他的图片检索系统中。
每张图片都有一个很长的相关描述，当用户们输入了一些关键词去查找图片，韦斯利的系统将会找出与用户输入的关键词匹配数最多的图片并把他显示给用户。
简单的说，给出一段描述图片的文字，再给你一些关键词，你要回答这段文字中匹配了多少个关键词。

输入格式

第一行一个整数n,表示有n个关键词(n<=10000)
接下来n行，每行表示一个关键词，每个关键词由小写字母构成，且长度不超过50
最后一行，表示描述图片的文字，由小写字母构成且长度不超过1000000

输出格式

一个整数，表示最多匹配的关键词数

样例输入

5
she
he
say
shr
her
yasherhs

样例输出

3