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;
}