陈老师神题
考虑怎么删除,每次插入最多增加两个节点,那么删除只要删除两个节点就好了
加入字符时会改变两个值,一个是nxt一个是fail
当我们删除节点 u u 时,如果一个点 的fail是 u u ,那么要把 改成 failu f a i l u ,这个可以用并查集实现
还有一个是nxt,nxt改变只有在增加nq的时候,可以用链表维护(大概也是个并查集)
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;
const int N=200010;
int n;
char a[N];
int len[N],fail[N],nxt[N][30],pos[N],S[N],vis[N],size[N],fa[N][20],ad[N],top,cnt=1;
ll ans;
inline int calc(int x){
int cur=x;
for(int i=17;~i;i--)
if(!vis[fa[x][i]]) x=fa[x][i];
return len[cur]-len[fa[x][0]];
}
int Find(int x){
if(!vis[fail[x]]) return fail[x]=Find(fail[x]);
return fail[x];
}
int Link[N];
int Get(int &x){
if(!vis[x] && x) return x=Get(Link[x]);
return x;
}
inline void extend(int p,int c,int ps){
int np=++cnt; len[np]=len[p]+1; pos[ps]=np; vis[np]=1;
while(p && !Get(nxt[p][c])) nxt[p][c]=np,p=fail[p];
if(!p) fail[np]=1,size[1]++;
else{
int q=nxt[p][c];
if(len[q]==len[p]+1) fail[np]=q,size[q]++;
else{
int nq=++cnt; len[nq]=len[p]+1; ad[ps]=nq; vis[nq]=1;
memcpy(nxt[nq],nxt[q],sizeof(nxt[nq]));
ans-=len[q]-len[Find(q)];
fail[nq]=fail[q]; Link[nq]=q;
fail[q]=fail[np]=nq;
ans+=len[q]-len[nq]+len[nq]-len[Find(nq)];
size[nq]+=2;
while(p && Get(nxt[p][c])==q) nxt[p][c]=nq,p=fail[p];
}
}
ans+=len[np]-len[Find(np)];
}
void PutAns(ll x){
if(x>=10) PutAns(x/10); putchar(x%10+'0');
}
inline void del(int x){
ans-=len[x]-len[Find(x)];
vis[x]=0; ans+=(len[x]-len[Find(x)])*size[x];
size[Find(x)]+=size[x]-1;
}
int main(){
scanf("%s",a+1); n=strlen(a+1);
pos[0]=1; vis[1]=1;
for(int i=1;i<=n;i++){
if(a[i]!='-'){
S[++top]=i; extend(pos[S[top-1]],a[i]-'a',i);
}
else{
int x=S[top--];
if(ad[x]) del(ad[x]);
del(pos[x]);
}
PutAns(ans); putchar('\n');
}
return 0;
}