1. 程式人生 > >bzoj2555(後綴自動機+LCT)

bzoj2555(後綴自動機+LCT)

for hup 出現 ins 必須 void pan += bzoj

題目描述

(1):在當前字符串的後面插入一個字符串 (2):詢問字符串s在當前字符串中出現了幾次?(作為連續子串) 你必須在線支持這些操作。 題解 做法很自然,建出後綴自動機,維護每個節點的right集合,對於詢問直接在sam上跑就好了。 然後它是在線的,得用LCT維護。 然後細節極多,首先必須維護好樹的形態,也就是說不能makeroot,所以我的link就長這樣。
inline void link(int x,int y){
    access(x);splay(x);access(y);splay(y);
    f[x]=y;si[y]+=size[x];pushup(y);
}

然後cut長這樣

inline void cut(int x,int y){
    access(x);splay(x);
    f[tr[x][0]]=0;tr[x][0]=0;
    pushup(x);
}

所以在這顆LCT中,每個點的左兒子維護的是它的父親,然後我們在算子樹和的時候把左兒子扣掉就好了。

斷斷續續淦了三天,極菜。

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 1300002
using namespace std;
int size[N],ch[N][26],tr[N][2],fa[N],si[N],last,cnt,n,l[N],q,f[N],val[N];
int mark; char s[N],qs[10]; inline void pushup(int x){size[x]=size[tr[x][0]]+size[tr[x][1]]+si[x]+val[x];} inline bool ge(int x){return tr[f[x]][1]==x;} inline bool isroot(int x){return tr[f[x]][0]!=x&&tr[f[x]][1]!=x;} inline void rotate(int x){ int y=f[x],o=ge(x); if(isroot(x))return; tr[y][o]
=tr[x][o^1];f[tr[y][o]]=y; if(!isroot(y))tr[f[y]][ge(y)]=x;f[x]=f[y]; f[y]=x;tr[x][o^1]=y;pushup(y);pushup(x); } inline void splay(int x){ while(!isroot(x)){ int y=f[x]; if(isroot(y))rotate(x); else rotate(ge(x)==ge(y)?y:x),rotate(x); } } inline void access(int x){ for(int y=0;x;y=x,x=f[x]){ splay(x); si[x]+=size[tr[x][1]];si[x]-=size[y];tr[x][1]=y; pushup(x); } } inline void link(int x,int y){ access(x);splay(x);access(y);splay(y); f[x]=y;si[y]+=size[x];pushup(y); } inline void cut(int x,int y){ access(x);splay(x); f[tr[x][0]]=0;tr[x][0]=0; pushup(x); } void Decode(char *ch,int mask){ int l=strlen(ch+1); for(int i=0;i<l;++i){ mask=(mask*131+i)%l; swap(ch[i+1],ch[mask+1]); } } inline void ins(int x){ int p=last,np=++cnt;last=np;l[np]=l[p]+1;size[np]=val[np]=1; for(;p&&!ch[p][x];p=fa[p])ch[p][x]=np; if(!p)fa[np]=1,link(np,1); else{ int q=ch[p][x]; if(l[p]+1==l[q])fa[np]=q,link(np,q); else{ int nq=++cnt;l[nq]=l[p]+1; memcpy(ch[nq],ch[q],sizeof(ch[q])); int y=fa[q];cut(q,y);link(nq,y); fa[nq]=fa[q];fa[q]=fa[np]=nq; link(q,nq);link(np,nq); for(;ch[p][x]==q;p=fa[p])ch[p][x]=nq; } } } int query(int n){ int now=1; for(int i=1;i<=n;++i)if(ch[now][s[i]-A])now=ch[now][s[i]-A];else return 0; access(now);splay(now); return size[now]-size[tr[now][0]]; } int main(){ scanf("%d",&q); scanf("%s",s+1);n=strlen(s+1);last=cnt=1; for(int i=1;i<=n;++i)ins(s[i]-A); while(q--){ scanf("%s",qs);scanf("%s",s+1);n=strlen(s+1); Decode(s,mark); if(qs[0]==Q){ int x=query(n);mark^=x;printf("%d\n",x); } else for(int i=1;i<=n;++i)ins(s[i]-A); } return 0; }

bzoj2555(後綴自動機+LCT)