标签：ios main 记录 bsp 自动机 出现 字符串 传送门 +=
我们正常的建好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[5000010][26], fail[5000010], sum[5000010], tot = 1; int n; int end_id[210]; char s[5000010]; queue<int> q; vector<int> vec[5000010]; void solve(int x) { for (int i = 0; i < (int)vec[x].size(); i++) { solve(vec[x][i]); sum[x] += sum[vec[x][i]]; } } int main() { cin >> n; for (int i = 1, p, len; i <= n; i++) { scanf("%s", s + 1); p = 1; len = strlen(s + 1); for (int j = 1; 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 = 0; i < 26; i++) trie[0][i] = 1; fail[1] = 0; q.push(1); while (!q.empty()) { int x = q.front(); q.pop(); for (int i = 0; i < 26; 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(1); for (int i = 1; i <= n; i++) { cout << sum[end_id[i]] << "\n"; } return 0; }
标签：ios main 记录 bsp 自动机 出现 字符串 传送门 +=
原文地址：https://www.cnblogs.com/zcr-blog/p/12322190.html
