BZOJ 3172 【AC自动机】单词

题目描述

某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

 

输入

第一个一个整数N,表示有多少个单词，接下来N行每行一个单词。每个单词由小写字母组成，N<=200,单词长度不超过10^6

 

输出

输出N个整数，第i行的数字表示第i个单词在文章中出现了多少次。

 

样例输入

3
a
aa
aaa

 

样例输出

6
3
1

AC自动机裸题，唯一麻烦的是记录出现次数时要遍历队列，只好手写一个队列

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1000005;
int q[maxn];
int trie[maxn][26];
int fail[maxn];
string str[205];
int cnt[205];
int sum[maxn];
int tot = 1;
void insert(int j)
{
    int l = str[j].length();
    int tmp = 0;
    for (int i = 0; i < l; i++)
    {
        int tmpp = str[j][i] - 'a';
        if (!trie[tmp][tmpp])
            trie[tmp][tmpp] = tot++;
        tmp = trie[tmp][tmpp];
        sum[tmp]++;
    }
    cnt[j] = tmp;
}
void getfail()
{
    int head,tail;
    head = tail = 0;
    for (int i = 0; i < 26; i++)
    {
        if (trie[0][i])
        {
            q[tail++] = trie[0][i];
            fail[trie[0][i]] = 0;
        }
    }
    while (head != tail)
    {
        int p = q[head];
        head++;
        for (int i = 0; i < 26; i++)
        {
            if (trie[p][i])
            {
                fail[ trie[p][i] ] = trie[ fail[p] ][i];
                q[tail++] = trie[p][i];
            }
            else
                trie[p][i] = trie[ fail[p] ][i];
        }
    }
    for (int i = tail; i >= 1; i--)
        sum[ fail[q[i]] ] += sum[ q[i] ];
}
int main()
{
//    freopen("in.txt", "r", stdin);
    int n;
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
    {
        cin >> str[i];
        insert(i);
    }
    getfail();
    for (int i = 0; i < n; i++)
        printf("%d\n", sum[ cnt[i] ]);
    return 0;
}