bzoj 4516 后缀自动机

最新推荐文章于 2019-09-12 22:25:01 发布

原创最新推荐文章于 2019-09-12 22:25:01 发布 · 187 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#ACM

后缀自动机专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种基于AC自动机的数据结构实现方法，通过在字符串后逐个添加字符并更新自动机状态来统计特定子串出现的次数。具体实现了节点的插入操作，并通过维护每个节点的深度来快速计算新增子串的数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在字符串后面每加一个字符，增加的以它为末尾的字符串个数为

mx[p]−mx[fa[p]]

#include <cstdio>
#include <map>
#include <algorithm>
using namespace std;
#define N 200010
#define ll long long
int root,last,n,cnt=0,fa[N],mx[N];
ll ans=0;
map<int,int>son[N]; 
inline void ins(int ch){
    int p=last,np=++cnt;last=np;mx[np]=mx[p]+1;
    while(p&&!son[p][ch]) son[p][ch]=np,p=fa[p];
    if(!p) fa[np]=root;
    else{
        int q=son[p][ch];
        if(mx[q]==mx[p]+1) fa[np]=q;
        else{
            int nq=++cnt;mx[nq]=mx[p]+1;son[nq]=son[q];
            fa[nq]=fa[q];fa[q]=fa[np]=nq;
            while(son[p][ch]==q) son[p][ch]=nq,p=fa[p];
        }
    }ans+=(ll)mx[np]-(ll)mx[fa[np]];
}
int main(){
    scanf("%d",&n);
    last=root=++cnt;
    for(int i=1;i<=n;i++){
        int x;scanf("%d",&x);
        ins(x);printf("%lld\n",ans);
    }
    return 0;
}