单词「TJOI 2013」(AC自动机)

传送门

我们正常的建好Trie后求一遍fail。之后对于每一个节点，从它的fail连向它一条单项边。然后从根节点开始dfs。

记sum[i]代表从根到i号节点所代表的的字符串出现的次数，即该点的权值。

设当前的节点为x，他有一个孩子y，则使sum[x] += sum[y]。

记得记录一下每个字符串结尾的节点编号，设第i个字符串结尾的编号为id[i]，对于每个字符串i最后输出sum[id[i]]即可。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
using namespace std;
int trie[][], fail[], sum[], tot = ;
int n;
int end_id[];
char s[];
queue<int> q;
vector<int> vec[];
void solve(int x) {
    for (int i = ; i < (int)vec[x].size(); i++) {
        solve(vec[x][i]);
        sum[x] += sum[vec[x][i]];
    }
}
int main() {
    cin >> n;
    for (int i = , p, len; i <= n; i++) {
        scanf("%s", s + );
        p = ;
        len = strlen(s + );
        for (int j = ; j <= len; j++) {
            int k = s[j] - 'a';
            sum[p]++;
            if (!trie[p][k]) trie[p][k] = ++tot;
            p = trie[p][k];
        }
        sum[p]++;
        end_id[i] = p;
    }
    for (int i = ; i < ; i++) trie[][i] = ;
    fail[] = ;
    q.push();
    while (!q.empty()) {
        int x = q.front();
        q.pop();
        for (int i = ; i < ; i++) {
            if (!trie[x][i]) {
                trie[x][i] = trie[fail[x]][i];
            } else {
                vec[trie[fail[x]][i]].push_back(trie[x][i]);
                fail[trie[x][i]] = trie[fail[x]][i];
                q.push(trie[x][i]);
            }
        }
    }
    solve();
    for (int i = ; i <= n; i++) {
        cout << sum[end_id[i]] << "\n";
    }
    return ;
}