【COGS2622】后缀平衡树

这是个后缀平衡树的裸题。。。。

然后傻逼的我调了一下午。

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
const int N=1e5+;
const int bas=;
int hs[N],M[N];
int n,len,ans,Ans[N];
inline int read(){
    int f=,x=;char ch;
    do{ch=getchar();if(ch=='-')f=-;}while(ch<''||ch>'');
    do{x=x*+ch-'';ch=getchar();}while(ch>=''&&ch<='');
    return f*x;
}
struct Suffix_Balanced_ScapeGoat_Tree{
    int lx[N],rx[N],rt,size[N],s[N],cnt,Q[N],tail;
    ll tag[N];
    int get(int i,int l){return hs[i]-hs[i-l]*M[l];}
    int qlcp(int x,int y){
        int l=,r=min(x,y);
        while(l<r){
            int mid=(l+r)>>;
            if(get(x,mid+)==get(y,mid+))l=mid+;else r=mid;
        }
        return l;
    }
    inline bool cmp(int x,int y){int l=qlcp(x,y);return s[x-l]<s[y-l];}
    inline int merge(int x,int y){
        if(!x||!y)return x|y;
        if(size[x]>size[y]){size[x]+=size[y];rx[x]=merge(rx[x],y);return x;}
        else{size[y]+=size[x];lx[y]=merge(x,lx[y]);return y;}
    }
    inline int build(int ls,int rs,ll l,ll r){
        if(ls>rs)return ;
        int mid=(ls+rs)>>;ll midv=(l+r)>>;int x=Q[mid];tag[x]=midv;
        lx[x]=build(ls,mid-,l,midv);rx[x]=build(mid+,rs,midv,r);
        size[x]=rs-ls+;return x;
    }
    inline void dfs(int x){if(lx[x])dfs(lx[x]);Q[++tail]=x;if(rx[x])dfs(rx[x]);}
    inline int rebuild(int x,ll l,ll r){
        tail=;dfs(x);return build(,tail,l,r);
    }
    inline int ins(int x,ll l,ll r,int val){
        if(!x){
            size[++cnt]=;lx[cnt]=rx[cnt]=;tag[cnt]=(l+r)>>;
            return cnt;
        }
        size[x]++;
        if(cmp(x,val)){
            rx[x]=ins(rx[x],tag[x],r,val);
            if(size[rx[x]]>0.65*size[x])x=rebuild(x,l,r);
        }
        else{
            lx[x]=ins(lx[x],l,tag[x],val);
            if(size[lx[x]]>0.65*size[x])x=rebuild(x,l,r);
        }
        return x;
    }
    inline int del(int x,int val){
        if(x==val)return merge(lx[x],rx[x]);
        size[x]--;
        if(tag[x]<tag[val])rx[x]=del(rx[x],val);
        else lx[x]=del(lx[x],val);
        return x;
    }
    inline int queryrk(int key){
        int x=rt,ans=;
        while(){
            int i=size[lx[x]]+;
            if(key==x)return ans+i;
            if(tag[x]<tag[key])x=rx[x],ans+=i;else x=lx[x];
        }
    }
    inline int find(int key){
        int x=rt;
        while(){
            int i=size[lx[x]]+;
            if(key==i)return x;
            if(key>i)x=rx[x],key-=i;else x=lx[x];
        }
    }
    inline void del(int x){
        int rk=queryrk(x),y=find(rk-),z=find(rk+);
        ans-=qlcp(x,y)+qlcp(x,z)-qlcp(y,z);
        rt=del(rt,x);cnt--;len--;
    }
    inline void ins(int x){
        s[++len]=x;hs[len]=hs[len-]*bas+x;
        rt=ins(rt,,1LL<<,len);
        if(len<)return;
        int rk=queryrk(len),y=find(rk-),z=find(rk+);
        ans+=qlcp(len,y)+qlcp(len,z)-qlcp(y,z);
    }
}T;
char str[N];
struct Edge{int u,v,next;}G[N<<];int head[N],tot=;
inline void addedge(int u,int v){
    G[++tot].u=u;G[tot].v=v;G[tot].next=head[u];head[u]=tot;
    G[++tot].u=v;G[tot].v=u;G[tot].next=head[v];head[v]=tot;
}
inline void dfs(int u,int fa){
    T.ins(str[u]-'a'+);Ans[u]=(len-)*(len-)/-ans;
    for(int i=head[u];i;i=G[i].next){
        if(G[i].v!=fa)dfs(G[i].v,u);
    }
    T.del(len);
}
int main(){
    freopen("balsuffix.in","r",stdin);
    freopen("balsuffix.out","w",stdout);
    M[]=;
    for(int i=;i<N;i++)M[i]=M[i-]*bas;
    T.ins();T.ins();int T=read();
    while(T--){
        n=read();for(int i=;i<=n;i++)head[i]=;tot=;
        for(int i=;i<n;i++){
            int u=read(),v=read();addedge(u,v);
        }
        scanf("%s",str+);
        dfs(,);
        for(int i=;i<=n;i++)printf("%d\n",Ans[i]);
    }
}