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

传送门

比较好的$Sam$简单题了

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

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

其实就是每次把修改或者新建了的节点u的$len[u]-len[link[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