AC自动机可以看作是在Trie树上建立了fail指针，在这里可以看作fail链。如果u的fail链指向v，那么v的对应串一定是u对应串在所给定字符串集合的后缀集合中的最长的后缀。
我们考虑一下如何实现这个东西。
以上数组实现的过程中我们让0号结点充当了根，这样会省去很多边界的处理。
我们考虑如何用指针去实现AC自动机，这看起来要更加自然。在指针实现的过程中，为了严谨起见，我们定义：节点的ch指针在默认情况下为NULL，fail指针在默认情况下为root（根节点自身除外）
这样的话就更加严格，各种边界情况都非常清晰。
#10057. 「一本通 2.4 例 1」Keywords Search
#include <bits/stdc++.h>
using namespace std;
namespace ACA {
struct Node {
Node *ch[26], *fail;
int val;
// Default value: fail=root ch=NULL
Node* clear(Node* defFail) {
fail = defFail;
for (int i = 0; i < 26; i++) ch[i] = 0;
val = 0;
return this;
}
};
Node *root = NULL, pool[1000005];
int index;
Node* newnode() { return pool[index++].clear(root); }
void insert(string str) {
Node* p = root;
for (int i = 0; i < str.length(); i++) {
int v = str[i] - 'a';
if (!p->ch[v])
p->ch[v] = newnode();
p = p->ch[v];
}
p->val++;
}
void build() {
queue<Node*> q;
for (int i = 0; i < 26; i++)
if (root->ch[i])
q.push(root->ch[i]);
while (!q.empty()) {
Node* p = q.front();
q.pop();
for (int i = 0; i < 26; i++) {
if (p->ch[i] != NULL)
p->ch[i]->fail = (p->fail->ch[i]) ? ((p->fail->ch[i])) : root, q.push(p->ch[i]);
else
p->ch[i] = p->fail->ch[i];
}
}
}
int query(string str) {
Node* p = root;
int ans = 0;
for (int i = 0; i < str.length(); i++) {
p = p->ch[str[i] - 'a'];
if (p == NULL)
p = root; // Attention!
for (Node* t = p; t != root && ~(t->val); t = t->fail) ans += t->val, t->val = -1;
}
return ans;
}
void solve() {
int n;
cin >> n;
index = 0;
root = newnode();
for (int i = 1; i <= n; i++) {
string tmp;
cin >> tmp;
insert(tmp);
}
build();
string art;
cin >> art;
cout << query(art) << endl;
}
} // namespace ACA
int main() {
int T;
// ios::sync_with_stdio(false);
cin >> T;
while (T--) ACA::solve();
}
#10058. 「一本通 2.4 练习 1」玄武密码
我们考虑query()中的那个while循环，它沿着当前节点到根的fail链跳，跳到不能再跳或者遇到已经处理过的节点为止。所谓处理节点，相当于标记这个节点对应的串已经被匹配。根据这个特性很容易构造出算法。
#include <bits/stdc++.h>
using namespace std;
namespace ACA {
struct Node {
Node *ch[4], *fail;
int val;
// Default value: fail=root ch=NULL
Node* clear(Node* defFail) {
fail = defFail;
for (int i = 0; i < 4; i++) ch[i] = 0;
val = 0;
return this;
}
};
Node *root = NULL, pool[40000005];
int index;
Node* newnode() { return pool[index++].clear(root); }
void insert(string str) {
Node* p = root;
for (int i = 0; i < str.length(); i++) {
int v = str[i];
if (!p->ch[v])
p->ch[v] = newnode();
p = p->ch[v];
p->val = max(p->val, i);
}
}
void build() {
queue<Node*> q;
for (int i = 0; i < 4; i++)
if (root->ch[i])
q.push(root->ch[i]);
while (!q.empty()) {
Node* p = q.front();
q.pop();
for (int i = 0; i < 4; i++) {
if (p->ch[i] != NULL)
p->ch[i]->fail = (p->fail->ch[i]) ? ((p->fail->ch[i])) : root, q.push(p->ch[i]);
else
p->ch[i] = p->fail->ch[i];
}
}
}
void query(string str) {
Node* p = root;
for (int i = 0; i < str.length(); i++) {
p = p->ch[str[i]];
if (p == NULL)
p = root; // Attention!
for (Node* t = p; t != root && ~(t->val); t = t->fail) t->val = -1;
}
}
int getans(string str) {
Node* p = root;
for (int i = 0; i < str.length(); i++) {
p = p->ch[str[i]];
if (p->val >= 0)
return i;
}
return str.length();
}
string pat[100005];
void solve() {
int n;
cin >> n;
cin >> n;
index = 0;
root = newnode();
string art;
cin >> art;
for (int j = 0; j < art.length(); j++) {
if (art[j] == 'S')
art[j] = 0;
if (art[j] == 'N')
art[j] = 1;
if (art[j] == 'W')
art[j] = 2;
if (art[j] == 'E')
art[j] = 3;
}
for (int i = 1; i <= n; i++) {
string tmp;
cin >> tmp;
for (int j = 0; j < tmp.length(); j++) {
if (tmp[j] == 'S')
tmp[j] = 0;
if (tmp[j] == 'N')
tmp[j] = 1;
if (tmp[j] == 'W')
tmp[j] = 2;
if (tmp[j] == 'E')
tmp[j] = 3;
}
insert(tmp);
pat[i] = tmp;
}
build();
query(art);
for (int i = 1; i <= n; i++) cout << getans(pat[i]) << endl;
}
} // namespace ACA
int main() {
ios::sync_with_stdio(false);
int T;
ACA::solve();
}
#10059. 「一本通 2.4 练习 2」Censoring
之前KMP专题里面出现过一个censoring的题目，只不过在这里我们把单串换成了多串，但思想是一致的。
搞一个输出栈，每次删除时候弹栈，栈中同时记录一下匹配到的节点即可。
注意到这里由于子串不会互相包含，所以query中的while就可以去掉了。
写的时候忘记给staj[0]赋值T掉了，要注意边界
#include <bits/stdc++.h>
using namespace std;
namespace ACA {
struct Node {
Node *ch[26], *fail;
int val;
// Default value: fail=root ch=NULL
Node* clear(Node* defFail) {
fail = defFail;
for (int i = 0; i < 26; i++) ch[i] = 0;
val = 0;
return this;
}
};
Node *root = NULL, pool[10000005];
int index;
int sta[5000005], top = 0;
Node* stap[5000005];
Node* newnode() { return pool[index++].clear(root); }
void insert(string& str) {
Node* p = root;
for (int i = 0; i < str.length(); i++) {
int v = str[i] - 'a';
if (!p->ch[v])
p->ch[v] = newnode();
p = p->ch[v];
}
p->val = str.length();
}
void build() {
queue<Node*> q;
for (int i = 0; i < 26; i++)
if (root->ch[i])
q.push(root->ch[i]);
while (!q.empty()) {
Node* p = q.front();
q.pop();
for (int i = 0; i < 26; i++) {
if (p->ch[i] != NULL)
p->ch[i]->fail = (p->fail->ch[i]) ? ((p->fail->ch[i])) : root, q.push(p->ch[i]);
else
p->ch[i] = p->fail->ch[i];
}
}
}
int query(string& str) {
Node* p = root;
int ans = 0;
stap[0] = root; // Warning!!!
for (int i = 0; i < str.length(); i++) {
p = p->ch[str[i] - 'a'];
if (p == NULL)
p = root; // Attention!
++top;
sta[top] = i;
stap[top] = p;
if (p->val > 0) {
top -= p->val;
p = stap[top];
}
}
for (int i = 1; i <= top; i++) cout << str[sta[i]];
return ans;
}
void solve() {
string art;
cin >> art;
int n;
cin >> n;
index = 0;
root = newnode();
for (int i = 1; i <= n; i++) {
string tmp;
cin >> tmp;
insert(tmp);
}
build();
query(art);
}
} // namespace ACA
int main() {
ios::sync_with_stdio(false);
ACA::solve();
}
#10060. 「一本通 2.4 练习 3」单词
某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。
正常插入所有单词串，不过对每个位置都要打标记，这样相当于插入了所有的前缀。构建fail树完成后，对fail树求树上前缀和（自底向上），此后每个节点的值就是这个节点代表的串在整个字典中的出现次数。
#include <bits/stdc++.h>
using namespace std;
namespace ACA {
struct Node {
Node *ch[26], *fail;
int val, cnt;
vector<Node*> cf;
// Default value: fail=root ch=NULL
Node* clear(Node* defFail) {
fail = defFail;
for (int i = 0; i < 26; i++) ch[i] = 0;
val = 0;
cnt = 0;
return this;
}
};
Node *root = NULL, pool[3000005];
Node* epos[100005];
int index;
Node* newnode() { return pool[index++].clear(root); }
void insert(int id, string str) {
Node* p = root;
for (int i = 0; i < str.length(); i++) {
int v = str[i] - 'a';
if (!p->ch[v])
p->ch[v] = newnode();
p = p->ch[v];
p->val++;
}
epos[id] = p;
}
void build() {
queue<Node*> q;
vector<Node*> tq;
for (int i = 0; i < 26; i++)
if (root->ch[i])
q.push(root->ch[i]), tq.push_back(root->ch[i]);
while (!q.empty()) {
Node* p = q.front();
q.pop();
for (int i = 0; i < 26; i++) {
if (p->ch[i] != NULL)
p->ch[i]->fail = (p->fail->ch[i]) ? ((p->fail->ch[i])) : root, q.push(p->ch[i]),
tq.push_back(p->ch[i]);
else
p->ch[i] = p->fail->ch[i];
}
}
for (int i = tq.size() - 1; i >= 0; --i)
if (tq[i]->fail)
tq[i]->fail->val += tq[i]->val;
}
string s[1005];
void solve() {
int n;
cin >> n;
index = 0;
root = newnode();
for (int i = 1; i <= n; i++) {
string tmp;
cin >> tmp;
insert(i, tmp);
s[i] = tmp;
}
build();
for (int i = 1; i <= n; i++) cout << epos[i]->val << endl;
}
} // namespace ACA
int main() {
ios::sync_with_stdio(false);
ACA::solve();
}
#10061. 「一本通 2.4 练习 4」最短母串
输入N个串，找到它们的最短母串，输出可行解中字典序最小的。
第一次写这种题真的卡了很久。
我们最终应当是在AC自动机上跑记忆化BFS，状态的表示上需要借助状态压缩。
建树的时候，对于每一条fail边，假设从u指向v，即v是u的后缀，那么如果v是一个单词的结尾，那么u也应当是。这样看来我们需要对节点自身的状态做一下fail树上自顶向下的前缀和。
搜索的时候，维护BFS队列，顺便记录每个状态的来源信息方便输出。
#include <bits/stdc++.h>
using namespace std;
const int N = 1000005;
int ch[N][26], fi[N], val[N], n, m, t1, t2, t3, t4, ind;
void ins(char *s, int id) {
int len = strlen(s), p = 0;
for (int i = 0; i < len; i++) {
if (ch[p][s[i] - 'A'] == 0)
ch[p][s[i] - 'A'] = ++ind;
p = ch[p][s[i] - 'A'];
}
val[p] |= 1 << (id - 1);
}
void build() {
queue<int> q;
for (int i = 0; i < 26; i++)
if (ch[0][i])
q.push(ch[0][i]);
while (!q.empty()) {
int p = q.front();
q.pop();
for (int i = 0; i < 26; i++)
if (ch[p][i])
fi[ch[p][i]] = ch[fi[p]][i], q.push(ch[p][i]);
else
ch[p][i] = ch[fi[p]][i];
int t = fi[p];
while (t && !val[t]) t = fi[t];
val[p] |= val[t];
}
}
char str[N];
struct Status {
int pos = 0, state = 0, cnt = 0;
Status(int a, int b, int c) : pos(a), state(b), cnt(c){};
};
int u[1005][1 << 12];
pair<int, int> fa[1005][1 << 12];
char o[1005][1 << 12];
string out;
int main() {
cin >> n;
for (int i = 1; i <= n; i++) cin >> str, ins(str, i);
build();
queue<Status> q;
q.push((Status){ 0, val[0], 0 });
while (!q.empty()) {
Status p = q.front();
q.pop();
int node = p.pos, cnt = p.cnt, state = p.state;
if (state == (1 << n) - 1) {
while (node) {
out += o[node][state];
pair<int, int> tmp = fa[node][state];
node = tmp.first;
state = tmp.second;
}
reverse(out.begin(), out.end());
cout << out << endl;
return 0;
}
for (int i = 0; i < 26; i++) {
int newstate = state | val[ch[node][i]];
int newnode = ch[node][i];
if (u[newnode][newstate])
continue;
u[newnode][newstate] = 1;
o[newnode][newstate] = i + 'A';
fa[newnode][newstate] = make_pair(node, state);
q.push((Status){ newnode, newstate, 0 });
}
}
}
#10062. 「一本通 2.4 练习 5」病毒
找到一个最小的不包含任何给定串的无限长串。
很显然这个串一定从某个位置开始循环。我们先对所有的终止节点打一个禁止标记，然后再fail树上跑自顶向下前缀和（因为终止节点再fail树上的所有孩子都应该被禁止），然后在所有没有被静止的节点间DFS，直到找到一条后向边。
#include <bits/stdc++.h>
using namespace std;
const int N = 1000005;
int ch[N][2], fail[N], val[N], u[N], n, m, t1, t2, t3, t4, ind;
vector<int> g[N];
void insert(char *s) {
int len = strlen(s), p = 0;
for (int i = 0; i < len; i++) {
if (ch[p][s[i] - '0'] == 0)
ch[p][s[i] - '0'] = ++ind;
p = ch[p][s[i] - '0'];
}
val[p]++;
}
void build() {
queue<int> q;
for (int i = 0; i < 2; i++)
if (ch[0][i])
q.push(ch[0][i]);
while (!q.empty()) {
int p = q.front();
q.pop();
for (int i = 0; i < 2; i++) {
if (ch[p][i])
fail[ch[p][i]] = ch[fail[p]][i], q.push(ch[p][i]);
else
ch[p][i] = ch[fail[p]][i];
}
val[p] |= val[fail[p]];
}
for (int i = 0; i <= ind; i++) {
// if(fail[i]) g[i].push_back(fail[i]);
g[i].push_back(ch[i][0]);
g[i].push_back(ch[i][1]);
}
}
bool dfs(int p) {
if (val[p])
return false;
// cout<<"dfs in "<<p<<endl;
u[p] = 1;
for (int i = 0; i < g[p].size(); i++) {
if (u[g[p][i]] == 2)
continue;
if (u[g[p][i]] == 1)
return true;
if (dfs(g[p][i]))
return true;
}
u[p] = 2; // cout<<"dfs out "<<p<<endl;
return false;
}
char str[N];
int main() {
ios::sync_with_stdio(false);
cin >> n;
for (int i = 1; i <= n; i++) cin >> str, insert(str);
build();
cout << (dfs(0) ? "TAK" : "NIE") << endl;
}
#10063. 「一本通 2.4 练习 6」文本生成器
老套路打禁止标记后做出自顶向下前缀和，然后暴力递推，设f[i][j]表示i步走到第j个节点，每次利用f[i][j]去刷所有f[i][j]的儿子，遇到禁止节点就将它的f[i][j]强行弄成0.最后求和即可。
#include <bits/stdc++.h>
using namespace std;
const int N = 10005;
int ch[N][26], fail[N], val[N], u[N], n, m, t1, t2, t3, t4, ind;
vector<int> g[N];
int f[N][105];
void insert(char *s) {
int len = strlen(s), p = 0;
for (int i = 0; i < len; i++) {
if (ch[p][s[i] - 'A'] == 0)
ch[p][s[i] - 'A'] = ++ind;
p = ch[p][s[i] - 'A'];
}
val[p]++;
}
void build() {
queue<int> q;
for (int i = 0; i < 26; i++)
if (ch[0][i])
q.push(ch[0][i]);
while (!q.empty()) {
int p = q.front();
q.pop();
for (int i = 0; i < 26; i++) {
if (ch[p][i])
fail[ch[p][i]] = ch[fail[p]][i], q.push(ch[p][i]);
else
ch[p][i] = ch[fail[p]][i];
}
val[p] |= val[fail[p]];
}
for (int i = 0; i <= ind; i++) {
for (int j = 0; j < 26; j++) g[i].push_back(ch[i][j]);
}
}
char str[N];
int main() {
ios::sync_with_stdio(false);
cin >> n >> m;
for (int i = 1; i <= n; i++) cin >> str, insert(str);
build();
f[0][0] = 1;
for (int j = 0; j <= m; j++) {
for (int i = 0; i <= ind; i++) {
if (val[i]) {
f[i][j] = 0;
continue;
}
for (int k = 0; k < g[i].size(); k++) {
(f[g[i][k]][j + 1] += f[i][j]) %= 10007;
}
}
}
int ans = 0;
for (int i = 0; i <= ind; i++) (ans += f[i][m]) %= 10007;
int tot = 1;
for (int i = 1; i <= m; i++) (tot *= 26) %= 10007;
cout << ((tot - ans) % 10007 + 10007) % 10007 << endl;
}
总结
AC自动机的这个专题是到目前为止做的最艰难的一个专题。原因在于一些基本的性质理解得不透彻，以及在指针实现的几道题目中因为对边界考虑得不够清楚导致各种诡异情况。
从实际角度来看，数组版本其实比较适合绝大多数题目。
应用方面，我们用AC自动机来解决的基本问题其实就是多串匹配，即给定一个单词本，看文章中出现了单词本中的多少个单词。这是很容易的。
如果我们要分别统计每个单词出现了多少次，就需要用到fail树上前缀和的处理技巧。我们知道，从某个节点沿着fail树走到根，经过的节点都是这个节点的后缀（这里我们已经用节点来直接指代节点对应的串），那么如果某个节点出现，它到根的fail链上所有的节点都应当已经出现，如果这些节点是某个单词的结尾，那么这个单词也应该被统计。因此，我们每次修改实际上修改的是fail树上某个点到根的一条链，我们很容易想到把这转化为在该点上打一个标记然后自底向上前缀和。
这个思路非常重要，练习3中我们要统计每个单词在单词本里以子串形式出现的次数时，我们用的就是这种思路。如果p点出现，那么显然fail[p],fail[fail[p]]都也出现。所以去做自底向上前缀和。
对于Censoring这样的问题，我们需要把输出构建成一个栈，并且维护栈中每个位置对应的当时所在自动机上节点的编号，这样如果弹出我们就可以很快的转移到弹出后的状态。
AC自动机上进行搜索和dp的难度就会稍高一些。总得来说，这类问题需要去记录状态，这个状态通常包含对节点的描述和对某些其它信息的描述。通常还需要结合树上前缀和。
#include <bits/stdc++.h>
using namespace std;
const int N = 1000005;
int ch[N][26], fi[N], val[N], n, m, t1, t2, t3, t4, ind;
void ins(char *s, int id) {
int len = strlen(s), p = 0;
for (int i = 0; i < len; i++) {
if (ch[p][s[i] - 'A'] == 0)
ch[p][s[i] - 'A'] = ++ind;
p = ch[p][s[i] - 'A'];
}
val[p] |= 1 << (id - 1);
}
void build() {
queue<int> q;
for (int i = 0; i < 26; i++)
if (ch[0][i])
q.push(ch[0][i]);
while (!q.empty()) {
int p = q.front();
q.pop();
for (int i = 0; i < 26; i++)
if (ch[p][i])
fi[ch[p][i]] = ch[fi[p]][i], q.push(ch[p][i]);
else
ch[p][i] = ch[fi[p]][i];
int t = fi[p];
while (t && !val[t]) t = fi[t];
val[p] |= val[t];
}
}
char str[N];
struct Status {
int pos = 0, state = 0, cnt = 0;
Status(int a, int b, int c) : pos(a), state(b), cnt(c){};
};
int u[1005][1 << 12];
pair<int, int> fa[1005][1 << 12];
char o[1005][1 << 12];
string out;
int main() {
cin >> n;
for (int i = 1; i <= n; i++) cin >> str, ins(str, i);
build();
queue<Status> q;
q.push((Status){ 0, val[0], 0 });
while (!q.empty()) {
Status p = q.front();
q.pop();
int node = p.pos, cnt = p.cnt, state = p.state;
if (state == (1 << n) - 1) {
while (node) {
out += o[node][state];
pair<int, int> tmp = fa[node][state];
node = tmp.first;
state = tmp.second;
}
reverse(out.begin(), out.end());
cout << out << endl;
return 0;
}
for (int i = 0; i < 26; i++) {
int newstate = state | val[ch[node][i]];
int newnode = ch[node][i];
if (u[newnode][newstate])
continue;
u[newnode][newstate] = 1;
o[newnode][newstate] = i + 'A';
fa[newnode][newstate] = make_pair(node, state);
q.push((Status){ newnode, newstate, 0 });
}
}
}