bzoj4516 [Sdoi2016]生成魔咒(SAM+map)

动态在最后字符,问本质不同的子串个数,可以SAM来做。
mx[x]-mn[x]+1=mx[x]-mx[par[x]]就是答案。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 200010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n=0,m,par[N],mx[N],last,rt;
ll ans=0;
map<int,int>son[N];
inline void ins(int ch){
    int p=last,np=++n;last=np;mx[np]=mx[p]+1;
    for(;p&&!son[p][ch];p=par[p]) son[p][ch]=np;
    if(!p){par[np]=rt;ans+=mx[np]-mx[par[np]];return;}
    int q=son[p][ch];
    if(mx[q]==mx[p]+1){par[np]=q;ans+=mx[np]-mx[par[np]];return;}
    int nq=++n;mx[nq]=mx[p]+1;son[nq]=son[q];
    par[nq]=par[q];par[q]=par[np]=nq;
    for(;p&&son[p][ch]==q;p=par[p]) son[p][ch]=nq;ans+=mx[np]-mx[par[np]];
}
int main(){
//  freopen("a.in","r",stdin);
    m=read();last=rt=++n;
    while(m--){
        int x=read();ins(x);printf("%lld\n",ans);
    }return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值