单词(AC自动机，拓扑排序)

题目链接

题意：

有一篇论文，是由很多的单词构成。但他发现一个单词会出现多次
现在有$N$个单词，问这个$N$个单词出现的次数

Sol：

建立Trie的同时，记录一个$f[i]$数组记录当前单词出现的次数（类似前缀和）。然后同样利用bfs构建$fail$数组，另外bfs的队列里面记录的节点顺序应该为Trie图的拓扑序列。
在所有单词中求某个单词的出现次数的时候，就是求所有前缀当中后缀出现的单词（$fail$数组的定义），最后就是从后往前累计单词的出现次数。

Code:

#include<bits/stdc++.h>
#define endl '\n' 
#define br putchar('\n') 
#define _x fixed << setprecision
#define debug(x) cerr<< #x << '=' << x << '\n'
#define ok() cout << "Yes\n" 
#define gg() cout << "No\n" 
#define mem(a, b) memset(a, b, sizeof(a));
#define rep(i, a, b) for (int i = a;i <= b; ++i)
#define rrep(i, a, b) for(int i = a; i >= b; --i)
#define ALL(a) (a).begin(), (a).end()
#define IOS ios::sync_with_stdio(false); cin.tie(0);cout.tie(0)
using namespace std;
typedef pair<int, int> PII;
const int mod = 1e9 + 7;
int dx[] = {-1, 0, 1, 0, -1, 1, 1, -1}, dy[] = {0, 1, 0, -1, 1, 1, -1, -1};

const int N = 1000010;

int trie[N][26], f[N], idx, id[300];
char str[N];
int q[N], fail[N];
int n;

void insert(int x)
{
    int p = 0;
    for(int i = 0; str[i]; ++ i)
    {
        int u = str[i] - 'a';
        if(!trie[p][u]) trie[p][u] = ++ idx;
        p = trie[p][u];
        f[p] ++;
    }
    id[x] = p;
}
// 构建fail数组，利用bfs从第二层开始，0是根节点
void build()
{
    int hh = 0, tt = -1;
    for(int i = 0; i < 26; ++i)// 将第一层所有入队
        if(trie[0][i]) q[++ tt] = trie[0][i];
    while(hh <= tt) // 进行bfs宽搜
    {
        int t = q[hh ++];
        for(int i = 0; i < 26; ++i)
        {
            if(trie[t][i]) { // 如果存在这个节点，就让他指向他父亲
            //节点t的fail指向的节点的具有相同字母(第i个)的位置
                fail[trie[t][i]] = trie[fail[t]][i];
                q[++ tt] = trie[t][i];
            }
            else
            {// 不存在就将这个节点指向这个节点的fail指向的节点具有相同
            // 字母的位置
                trie[t][i] = trie[fail[t]][i];
            }
        }
    }
}

signed main(){
#ifdef ONLINE_JUDGE
#else
    freopen("in.txt", "r", stdin);  
    freopen("out.txt", "w", stdout);
#endif
    IOS;
    cin >> n;
    for(int i = 0; i < n; ++i){
        cin >> str;
        insert(i);
    }
    cin >> str;
    build();
    for(int i = idx - 1; i >= 0; -- i) f[fail[q[i]]] += f[q[i]];

    for(int i = 0; i < n; ++i) cout << f[id[i]] << endl;
    return 0;
}