P3966 [TJOI2013]单词

链接

题目链接

题解

多模式匹配，很容易想到AC自动机。

首先只给出了模板串，可以将它们用一个字符集外的字符连接起来形成文本串，然后放到AC自动机里去匹配。

但是往上暴跳fail指针的复杂度是不对的，考虑这样优化：先建出fail树，然后匹配时记录相应的结点经过的次数，最后通过一次树形DP来统计。如果一个单词不是任何单词的前缀，那么它对应fail树上的一个叶子结点，答案就是该结点被经过的次数；如果是某些单词的前缀，那么它的子树和就是它被经过的次数。

事实上，并不需要搞出上面的文本串，把单词一个个放进去也是可以的。此外，这题还有SA+单调栈、SAM甚至哈希做法。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxlen = 1.5e6 + 9;
const int maxn = 509;
const ll mod = 998244353;

struct tnode
{
    int ch[27];
    int cnt = 0, fail = 0;
};
tnode trie[maxlen];
struct node
{
    int nxt, to;
};
node w[maxlen << 1];
int head[maxlen << 1], cnt, ans;
string str[maxn], s;
inline void add(int x, int y)
{
    w[++cnt].nxt = head[x];
    w[cnt].to = y;
    head[x] = cnt;
}
int pos[maxn], sum[maxlen << 1], n, num;
void ins(int x)
{
    int u = 0, len = str[x].length();
    for (int i = 0; i < len; ++i)
    {
        int c = str[x][i] - 'a';
        if (!trie[u].ch[c])
            trie[u].ch[c] = ++num;
        u = trie[u].ch[c];
    }
    trie[u].cnt++;
    pos[x] = u;
}
void build()
{
    queue<int> q;
    for (int i = 0; i < 26; ++i)
    {
        if (trie[0].ch[i])
        {
            trie[trie[0].ch[i]].fail = 0;
            q.push(trie[0].ch[i]);
        }
    }
    while (!q.empty())
    {
        int x = q.front();
        q.pop();
        for (int i = 0; i <= 26; ++i)
        {
            if (trie[x].ch[i])
            {
                trie[trie[x].ch[i]].fail = trie[trie[x].fail].ch[i];
                q.push(trie[x].ch[i]);
            }
            else
            {
                trie[x].ch[i] = trie[trie[x].fail].ch[i];
            }
        }
    }
}
void dfs(int x)
{
    for (int i = head[x]; i; i = w[i].nxt)
    {
        dfs(w[i].to);
        sum[x] += sum[w[i].to];
    }
}

signed main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);
    std::cout.tie(0);
    cin >> n;
    num = cnt = 0;
    s = "";
    for (int i = 1; i <= n; ++i)
    {
        cin >> str[i];
        s += str[i];
        s += (char)('z' + 1);
        ins(i);
    }
    build();
    for (int i = 1; i <= num; ++i)
        add(trie[i].fail, i);
    int u = 0, len = s.length();
    for (int i = 0; i < len; ++i)
    {
        int j = s[i] - 'a';
        u = trie[u].ch[j];
        sum[u]++;
    }
    dfs(0);
    for (int i = 1; i <= n; ++i)
        cout << sum[pos[i]] << '\n';
    return 0;
}