[后缀平衡树][JZOJ4384]hashit

题目描述

有一个字符串s，一开始为空串，要求支持两种操作：
1， ‘c’，在s后面加入小写字母字符c。
2， ‘-‘，删除最后一个字符。
问每次操作过后s有多少个两两不同的字符串。
操作序列也是一个字符串。
操作序列长度<=10^5

分析

当然可以离线后缀自动机乱搞…
在线做法就是后缀平衡树啦…只是把后缀数组在线而已，height用哈希来维护，就是裸题了。
后缀平衡树实际上跟后缀数组没有大区别，优化的插入就是把rank用重量平衡树维护。比如treap。
注意小细节：
删除的时候，是把点旋到叶子再删掉，旋转的时候记得实时维护root。

代码

#include<cstdio>#include<algorithm>#include<cstring>#include<cmath>#include<set>using namespace std;#define fo(i,j,k) for(i=j;i<=k;i++)#define fd(i,j,k) for(i=j;i>=k;i--)#define cmax(a,b) (a=(a>b)?a:b)#define cmin(a,b) (a=(a<b)?a:b)typedef long long ll;typedef double db;const int N=200005,M=1e6+5,mo=1e9+7,ws=1331;int tr[N][2],fa[N],key[N],height[N],lef[N],rig[N];int has[N],ss[N],d[N],n,len,i,rt,kan;ll rank[N],L[N],R[N],ans;char s[N],st[N];int w[20],I;void Print(ll x){    w[0]=0;    while (x) w[++w[0]]=x%10,x/=10;    if (!w[0]) putchar('0');    else fd(I,w[0],1) putchar('0'+w[I]);    putchar('\n');}void redo(int x,ll l,ll r){    if (!x) return ;    L[x]=l;    R[x]=r;    rank[x]=(l+r)/2;    redo(tr[x][0],l,rank[x]-1);    redo(tr[x][1],rank[x]+1,r);}int pd(int x){    return tr[fa[x]][1]==x;}void rotate(int x){    int y=fa[x],z=pd(x);    if (fa[y]) tr[fa[y]][pd(y)]=x;    fa[x]=fa[y];    if (tr[x][1-z]) fa[tr[x][1-z]]=y;    tr[y][z]=tr[x][1-z];    tr[x][1-z]=y;    fa[y]=x;    if (y==rt) rt=x;    redo(x,L[y],R[y]);}int cmp(int x,int y){    if (!(st[x]==st[y])) return st[x]<st[y];    return rank[x-1]<rank[y-1];}int hash(int x,int z){    return ((has[x]-1ll*has[x-z]*ss[z])%mo+mo)%mo;}void geth(int x,int y){    ans-=(y-height[y]);    int l=0,r=min(x,y),mid;    while (l<r)    {        mid=(l+r+1)/2;        if (hash(x,mid)==hash(y,mid)) l=mid;        else r=mid-1;    }    height[y]=l;    ans+=y-height[y];}int ins(int x,int y){    int tmp,z=cmp(x,y);// smaller 0, bigger 1    if (z) lef[y]=x;     else rig[y]=x;    if (!tr[x][z])    {        tr[x][z]=y;        fa[y]=x;        L[y]=L[x];R[y]=R[x];        if (!z) R[y]=rank[x]-1;        else L[y]=rank[x]+1;        rank[y]=(L[y]+R[y])/2;        ans+=y;        geth(lef[y],y);         geth(y,rig[y]);        tmp=y;    }else tmp=ins(tr[x][z],y);    if (fa[tmp]&&key[tmp]<key[fa[tmp]])        rotate(tmp);    else tmp=x;    return tmp;}void del(int x){    while (tr[x][0]||tr[x][1])     {        if (key[tr[x][1]]<key[tr[x][0]]&&tr[x][1]||(!tr[x][0]))            rotate(tr[x][1]);        else rotate(tr[x][0]);    }       if (!fa[x]) rt=0;else     {        tr[fa[x]][pd(x)]=0;            ans-=(x-height[x])+(rig[x]-height[rig[x]]);        lef[rig[x]]=lef[x];        rig[lef[x]]=rig[x];        cmin(height[rig[x]],height[x]);        ans+=rig[x]-height[rig[x]];    }    //tr[x][0]=tr[x][1]=fa[x]=height[x]=lef[x]=rig[x]=0;    //L[x]=R[x]=0;}int main(){    freopen("hashit.in","r",stdin);    freopen("hashit.out","w",stdout);    scanf("%s",s+1);    n=strlen(s+1);    ss[0]=1;    fo(i,1,n) ss[i]=1ll*ss[i-1]*ws%mo;    srand(n);    len=0;    fo(i,1,n)    {        if (s[i]=='-') del(len),len--;        else        {            len++;            height[len]=0;            has[len]=(1ll*has[len-1]*ws+s[i])%mo;            st[len]=s[i];            key[len]=1ll*rand()*rand()%mo;            if (len==1)            {                rt=1;                L[1]=1;                R[1]=1e16;                rank[1]=(L[1]+R[1])/2;                ans=1;            }else            {                rt=ins(rt,len);            }        }        Print(ans);    }}