【TJOI2013】单词

Description

给定若干模式串组成的文本串，求每个模式串出现的次数。

Solution

由于模式串可能重复，所以直接按照AC自动机的模板会出错，所以我们定义一个same数组，end数组只记录最早的编号，每次如果要覆盖，便把令该节点的same值为这个位置上的end的值，查询一遍，输出答案就输出ans[same[i]]。

我们再定义两个数组vis和sum表示当前节点有没有被扫过，当前节点被覆盖的次数。那么在统计答案的时候，如果当前节点是某个模式串的结尾并且没有扫过，那么说明这个模式串有贡献，就将答案+sum[now],之后将vis赋值为1，这样跑一边AC自动机就可以了。

Code

#include <bits/stdc++.h>
using namespace std;
int n, tot, same[100010], ans[100010], ch[500010][26], end[500010], vis[500010], sum[500010];
int f[500010];
char c[210][100010];
inline int read() {
    int ret = 0, op = 1;
    char c = getchar();
    while (!isdigit(c)) {
        if (c == '-') op = -1; 
        c = getchar();
    }
    while (isdigit(c)) {
        ret = ret * 10 + c - '0';
        c = getchar();
    }
    return ret * op;
}
inline void write(int x) {
    if (x < 0) putchar('-'), x = -x;
    if (x >= 10) write(x / 10);
    putchar(x % 10 + '0');
}
void insert(char *a, int id) {
    int now = 0, l = strlen(a);
    for (int i = 0; i < l; ++i) {
        if (!ch[now][a[i] - 'a']) ch[now][a[i] - 'a'] = ++tot;
        now = ch[now][a[i] - 'a'];
        sum[now]++;
    }
    if (!end[now]) end[now] = id;
    else same[id] = end[now];
}
queue <int> q;
void bfs() {
    for (register int i = 0; i < 26; ++i)
        if (ch[0][i]) {
            f[ch[0][i]] = 0;
            q.push(ch[0][i]);
        }
    while (!q.empty()) {
        int now = q.front();
        q.pop();
        for (register int i = 0; i < 26; ++i) {
            if (ch[now][i]) f[ch[now][i]] = ch[f[now]][i], q.push(ch[now][i]);
            else ch[now][i] = ch[f[now]][i];
        }
    }
}
void query(char *a) {
    int now = 0;
    int l = strlen(a);
    for (int i = 0; i < l; ++i) {
        now = ch[now][a[i] - 'a'];
        for (register int j = now; j; j = f[j])
            if (end[j] && !vis[now]) ans[end[j]] += sum[now];
        vis[now] = 1;
    }
}
int main() {
    n = read();
    for (register int i = 1; i <= n; ++i) {
        scanf("%s", c[i]);
        insert(c[i], i);
    }
    bfs();
    for (register int i = 1; i <= n; ++i) 
        if (!same[i]) query(c[i]);
    for (register int i = 1; i <= n; ++i) {
        if (!same[i]) write(ans[i]);
        else write(ans[same[i]]);
        puts("");
    }
    return 0;
}