OI模板 AC自动机
trie 树上算失配指针,感觉和kmp没啥关系
简单版:
//P3808
#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
struct AhoCorasick_Automaton{
struct node{ int ch[26], end, fail; } T[N];
int cnt;
void init(){
for(int i = 0; i < N; ++ i){
memset(T[i].ch, 0, sizeof(T[i].ch));
T[i].end = T[i].fail = 0;
}
cnt = 0;
}
void ins(char *s){
int len = strlen(s), now = 0;
for(int i = 0; i < len; ++ i){
char c = s[i] - 'a';
if(!T[now].ch[c]) T[now].ch[c] = ++cnt;
now = T[now].ch[c];
}
++ T[now].end;
}
void build(){
queue<int> q;
for(int i = 0; i < 26; ++ i) if(T[0].ch[i]) q.push(T[0].ch[i]);
while(!q.empty()){
int now = q.front(); q.pop();
for(int i = 0; i < 26; ++ i){
if(T[now].ch[i]){
T[T[now].ch[i]].fail = T[T[now].fail].ch[i];
q.push(T[now].ch[i]);
} else T[now].ch[i] = T[T[now].fail].ch[i];
}
}
}
int query(char *s){
build();
int len = strlen(s), now = 0, ans = 0;
for(int i = 0; i < len; ++ i){
now = T[now].ch[s[i]-'a'];
for(int i = now; i && T[i].end != -1; i = T[i].fail)
ans += T[i].end, T[i].end = -1;
}
return ans;
}
} AC;
char s[N]; int n;
int main(){
AC.init();
scanf("%d",&n);
for(int i=1; i<=n; ++i)
scanf("%s",s),
AC.ins(s);
scanf("%s",s);
printf("%d",AC.query(s));
return 0;
}
增强版:
名义上的增强版,实际的简单版
可以直接重复访问 T[i].end
,虽然这样复杂度是错的,但是这题能过。
int query(char *s){
build();
int len = strlen(s), now = 0;
for(int i = 0; i < len; ++ i){
now = T[now].ch[s[i]-'a'];
for(int i = now; i; i = T[i].fail)
++ ans[T[i].end];
}
}
二次增强版:
复杂度错的怎么办?考虑仍然只遍历一次。
可以发现若开始跳 fail 一定是沿 fail 树往根跳,所以我们可以不跳,最后通过拓扑排序把答案传上去。
虽然题解说是个 DAG,但是我觉得还是棵树。
//P5357
#include <bits/stdc++.h>
using namespace std;
const int N = 2e6 + 10;
char s[N]; int n, ans[N];
struct Trie{
struct node{ int ch[26], end, val; } T[N];
int cnt;
void ins(char *s, int num){
int len = strlen(s), now = 0;
for(int i = 0; i < len; ++ i){
char c = s[i] - 'a';
if(!T[now].ch[c]) T[now].ch[c] = ++ cnt;
now = T[now].ch[c];
}
T[now].end = num;
}
int query(char *s){
int len = strlen(s), now = 0;
for(int i = 0; i < len; ++ i){
char c = s[i] - 'a';
if(!T[now].ch[c]) return 0;
now = T[now].ch[c];
}
return T[now].end;
}
} trie;
struct AhoCorasick_Automaton{
struct node{ int ch[26], end, fail, ans; } T[N];
int cnt, ind[N];
void ins(char *s, int num){
int len = strlen(s), now = 0;
for(int i = 0; i < len; ++ i){
char c = s[i] - 'a';
if(!T[now].ch[c]) T[now].ch[c] = ++cnt;
now = T[now].ch[c];
}
T[now].end = num;
}
void build(){
queue<int> q;
for(int i = 0; i < 26; ++ i) if(T[0].ch[i]) q.push(T[0].ch[i]);
while(!q.empty()){
int now = q.front(); q.pop();
for(int i = 0; i < 26; ++ i){
if(T[now].ch[i]){
T[T[now].ch[i]].fail = T[T[now].fail].ch[i];
q.push(T[now].ch[i]), ++ ind[T[T[now].fail].ch[i]];
} else T[now].ch[i] = T[T[now].fail].ch[i];
}
}
}
int query(char *s){
build();
int len = strlen(s), now = 0;
for(int i = 0; i < len; ++ i)
now = T[now].ch[s[i]-'a'], ++ T[now].ans;
queue<int> q;
for(int i = 0; i <= cnt; ++ i) if(!ind[i]) q.push(i);
while(!q.empty()){
int x = q.front(); q.pop(); ans[T[x].end] = T[x].ans;
int y = T[x].fail; -- ind[y], T[y].ans += T[x].ans;
if(!ind[y]) q.push(y);
}
}
} AC;
int main(){
scanf("%d", &n);
for(int i = 1; i <= n; ++ i){
scanf("%s", s);
int t = trie.query(s);
if(t) ans[i] = -t;
else trie.ins(s, i), AC.ins(s, i);
}
scanf("%s", s); AC.query(s);
for(int i = 1; i <= n; ++ i) printf("%d\n", ans[i]>=0 ? ans[i] : ans[-ans[i]]);
return 0;
}