bzoj4768: wxh loves substring //后缀平衡树

bzoj4768: 2555加强版之wxh loves substring

---

题意

给出一个字符串，要求资瓷：

在末尾添加/删除字符；
询问一个串的出现次数。

原串长与变化长度之和<=800000，询问串总长<=3000000。

---

题解

后缀平衡树的板子题。
刚开始以为后缀平衡树是个很厉害的suffix data structure…学完后感觉就是拿了个平衡树维护SA？
我选择sgt，好写。我可能不会treap
具体过程的话，添加字符就是插一个（反串的）后缀进去，比较的时候先比较第一个字符，再比较剩下那一段后缀。出现过的后缀大小比较可以通过标号做到O(1)。为了维护这个标号需要重量平衡树。查询的时候，答案是$rk(Q+CHAR \_ MAX)-rk(Q+CHAR\_ MIN)$
论文比我说得明白多了x

前一天对着数据调了很长时间还是WA，今天早上发现解密写错了

---

代码

#include<bits/stdc++.h>
#define N 800005
#define mid (l+r>>1)
#define upd(x) sz[x]=sz[c[x][0]]+sz[c[x][1]]+1
using namespace std;
typedef long long ll;
int n,l,q,msk,ans,rt,tot,
c[N][2],sz[N],st[N];
char S[N],Q[N*3],op[7];
ll no[N];
bool tp;
inline void rd()
{
    tot=msk;
    scanf("%s",Q),l=strlen(Q);
    for(int i=0;i<l;i++)
    swap(Q[i],Q[tot=(tot*131+i)%l]);
}
void Uni(int x,int y,int &p)
{
    p=x&&y?sz[x]>sz[y]?
    Uni(c[x][1],y,c[x][1]),upd(x),x:
    (Uni(x,c[y][0],c[y][0]),upd(y),y):(x+y);
}
void Del(int &x,int y)
{
    x^y?sz[x]--,Del(c[x][no[y]>no[x]],y):
    Uni(c[x][0],c[x][1],x);
}
inline int Rk()
{
    for(int i=rt;i;i=c[i][tp])
    {
        tp=0;
        for(int j=0;j<=l;j++)
        if(Q[j]^S[i-j])
        {tp=Q[j]>S[i-j];break;}
        tp?ans+=c[i][0][sz]+1:0;
    }
}
void Flt(int &x)
{x?Flt(c[x][0]),st[++tot]=x,Flt(c[x][1]),x=0:0;}
int Reb(int l,int r,ll L,ll R)
{
    int x=st[mid];
    sz[x]=r-l+1,no[x]=L+R>>1;
    l<mid?c[x][0]=Reb(l,mid-1,L,no[x]):0;
    mid<r?c[x][1]=Reb(mid+1,r,no[x],R):0;
    return x;
}
void Ins(int &x,int y,ll l,ll r,bool d)
{
    if(!x)x=y,c[x][0]=c[x][1]=0,sz[x]=1,no[x]=mid;
    else
    {
        sz[x]++;
        bool t1=S[x]==S[y]?no[x-1]<no[y-1]:S[x]<S[y],
        t2=d&&sz[x]*0.7<c[x][t1][sz]+1;
        Ins(c[x][t1],y,t1?no[x]:l,t1?r:no[x],d&!t2);
        t2?tot=0,Flt(x),x=Reb(1,tot,l,r):0;
    }
}
int main()
{
    scanf("%d",&q);
    scanf("%s",S+1),n=strlen(S+1);
    for(int i=1;i<=n;i++)
    Ins(rt,i,0,1e18,1);
    while(q--)
    {
        scanf("%s",op);
        if(op[0]=='D')
        {
            scanf("%d",&l),n-=l;
            for(int i=1;i<=l;i++)
            Del(rt,n+i);
        }
        else if(op[0]=='A')
        {
            rd();
            for(int i=1;i<=l;i++)
            S[n+i]=Q[i-1],Ins(rt,n+i,0,1e18,1);
            n+=l;
        }
        else
        {
            rd(),ans=0;
            for(int i=0;i<l+1>>1;i++)
            swap(Q[i],Q[l-i-1]);
            Rk(),ans=-ans,Q[l]=127,
            Rk(),printf("%d\n",ans),msk^=ans;
        }
    }
}