Atcoder ABC268Ex Taboo 题解 [ 紫 ] [ AC 自动机 ]

Taboo：唐氏题，但是我忘了在 fail 树下传标记浪费了 0.5 h。

思路

显然，当我们匹配到一个字符串后，最优的做法是在结尾设置一个星号。因为放到后面能覆盖掉更多的区间，把模式串中匹配的子串尽可能多地破坏掉。

那么我们建出 ACAM 后跑多模式串匹配，遇到匹配的直接标记为星号，然后回到 ACAM 的根节点继续匹配即可。

注意一定要在 fail 树上下传匹配标记！

时间复杂度 \(O(n|\sum|)\)。

代码

#include <bits/stdc++.h>
#define fi first
#define se second
#define eb(x) emplace_back(x)
#define pb(x) push_back(x)
#define lc(x) (tr[x].ls)
#define rc(x) (tr[x].rs)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ldb;
using pi=pair<int,int>;
const int N=500005;
char s[N],t[N];
int ch[N][30],idx=0,ne[N],n;
bitset<N>cnt;
void insert(char *t)
{
    int p=0;
    for(int i=1;t[i];i++)
    {
        int j=t[i]-'a';
        if(ch[p][j]==0)ch[p][j]=++idx;
        p=ch[p][j];
    }
    cnt[p]=1;
}
void build()
{
    queue<int>q;
    for(int i=0;i<26;i++)
    {
        if(ch[0][i])q.push(ch[0][i]);
    }
    while(!q.empty())
    {
        int u=q.front();q.pop();
        cnt[u]=(cnt[u]|cnt[ne[u]]);
        for(int i=0;i<26;i++)
        {
            int v=ch[u][i];
            if(v)ne[v]=ch[ne[u]][i],q.push(v);
            else ch[u][i]=ch[ne[u]][i];
        }
    }
}
void solve()
{
    int p=0,ans=0;
    for(int i=1;s[i];i++)
    {
        int j=s[i]-'a';
        p=ch[p][j];
        if(cnt[p])p=0,ans++;
    }
    cout<<ans;
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin>>s+1;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>t+1;
        insert(t);
    }
    build();
    solve();
    return 0;
}