时间复杂度
时间复杂度是线性的,和所有单词的总长度有关,也就是\(O(n)\)。
#include <bits/stdc++.h>
using namespace std;
const int N = 1000010;
int n;
int tr[N][26], idx;
int f[N]; // 当前节点代表的字符串在整个trie中出现的次数,也用来记录递推结果
char s[N]; // 字符串
int id[210]; // 每个单词在trie中对应节点的编号,比如id[1]=2,表示第一个模式串,在trie树中是2号节点
void insert(int x) {
int p = 0;
for (int i = 0; s[i]; i++) {
int t = s[i] - 'a';
if (!tr[p][t]) tr[p][t] = ++idx;
p = tr[p][t];
f[p]++; //记录p节点代表的字符串在整个trie中出现的次数
}
id[x] = p; //记录x号单词在trie树中的节点编号
}
int q[N], ne[N];
void build() {
int hh = 0, tt = -1;
for (int i = 0; i < 26; i++)
if (tr[0][i]) q[++tt] = tr[0][i];
while (hh <= tt) {
int t = q[hh++];
for (int i = 0; i < 26; i++) {
if (!tr[t][i])
tr[t][i] = tr[ne[t]][i];
else {
ne[tr[t][i]] = tr[ne[t]][i];
q[++tt] = tr[t][i];
}
}
}
}
int main() {
scanf("%d", &n);
for (int i = 0; i < n; i++) {
scanf("%s", s);
//构建Trie树
insert(i);
}
//构建AC自动机
build();
//拓扑序更新
//从下向上递推更新, trie中节点编号为0~idx,一共idx+1个点,0既代表根节点又代表空节点
for (int i = idx; i; i--) f[ne[q[i]]] += f[q[i]];
//输出
for (int i = 0; i < n; i++) printf("%d\n", f[id[i]]);
return 0;
}