BZOJ 4516: [Sdoi2016]生成魔咒

/**************************************************************
    Problem: 4516
    User: BeiYu
    Language: C++
    Result: Accepted
    Time:812 ms
    Memory:14940 kb
****************************************************************/
 
#include<cstdio>
#include<cstring>
#include<map>
#include<iostream>
using namespace std;
 
typedef long long LL;
const int N = 200005;
 
int n,cnt,rt,lst;
LL ans;
map<int,int> go[N];
int par[N],val[N];
 
inline int in(int x=0,char ch=getchar()){ while(ch>‘9‘||ch<‘0‘) ch=getchar();
    while(ch>=‘0‘&&ch<=‘9‘) x=(x<<3)+(x<<1)+ch-‘0‘,ch=getchar();return x; }
 
void Extend(int w=in()){
    int p=lst,np=++cnt;
    val[np]=val[p]+1;
    while(p && go[p][w]==0) go[p][w]=np,p=par[p];
    if(!p) par[np]=rt;
    else{
        int q=go[p][w];
        if(val[p]+1 == val[q]) par[np]=q;
        else{
            int nq=++cnt;
            val[nq]=val[p]+1,go[nq]=go[q],par[nq]=par[q];
            par[q]=par[np]=nq;
            while(p && go[p][w]==q) go[p][w]=nq,p=par[p];
        }
    }ans+=val[np]-val[par[np]],lst=np;
}
 
int main(){
    lst=rt=++cnt,n=in();
    for(int i=1;i<=n;i++) Extend(),printf("%lld\n",ans);
    return 0;
}