【SDOI2016】—生成魔咒（后缀自动机）

最新推荐文章于 2022-04-05 17:04:07 发布

转载最新推荐文章于 2022-04-05 17:04:07 发布 · 102 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/stargazer-cyk/p/10366381.html

本文介绍了一个基于后缀自动机的算法，用于解决统计每加入一个字符会生成多少个新子串的问题。通过理解后缀自动机的工作原理，我们可以高效地更新和查询字符串的子串数量。

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

传送门

比较好的 $S a m$ 简单题了

主要看你对后缀自动机的理解

其实就是让你统计每加入一个字符会生成多少个新的子串

其实就是每次把修改或者新建了的节点u的 $l e n [u] - l e n [l i n k [u]]$ ，加起来就是了

#include<bits/stdc++.h>
using namespace std;
#define ll long long
inline int read(){
    char ch=getchar();
    int res=0;
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch))res=(res<<3)+(res<<1)+(ch^48),ch=getchar();
    return res;
}
struct node{
    int len,link;
    map<int,int>nxt;
};
#define max(a,b) (((a)>(b))?a:b)
const int N=100005;
node st[N<<1];
int tot,f[N<<1],cnt[N<<1],last;
ll ans;
char a[N];
inline ll calc(int x){
    return st[x].len-st[st[x].link].len;
}
inline void init(){
    tot=last=0;
    st[0].len=0,st[0].link=-1;
    ++tot;
}
inline void sa_extend(int c){
    int cur=tot++;
    st[cur].len=st[last].len+1;
    int p;
    for(p=last;p!=-1&&!st[p].nxt.count(c);p=st[p].link) st[p].nxt[c]=cur;
    if(p==-1)st[cur].link=0,ans+=calc(cur);
    else{
        int q=st[p].nxt[c];
        if(st[p].len+1==st[q].len)st[cur].link=q,ans+=calc(cur);
        else {
            int clo=tot++;
            st[clo].len=st[p].len+1;
            st[clo].nxt=st[q].nxt;
            st[clo].link=st[q].link;
            ans+=calc(clo),ans-=calc(q);           
            st[q].link=st[cur].link=clo;          
            ans+=calc(cur),ans+=calc(q);           
            for(;p!=-1&&st[p].nxt[c]==q;p=st[p].link)st[p].nxt[c]=clo;
        }
    }
    last=cur;
}
int main(){
    int n=read();init();
    for(int i=1;i<=n;i++)sa_extend(read()),cout<<ans<<'\n';
}

转载于:https://www.cnblogs.com/stargazer-cyk/p/10366381.html