可持久化平衡树

这里主要讲一下为什么 merge 函数需要新建节点的问题,以免以后忘掉。其他都是在分裂与更新的时候新建节点的问题。

可持久化平衡树

这张图片是我们执行分裂后,\(9,4\) 分别是我们分裂出来的两颗子树。

然后我们考虑对 \(9,4\) 执行合并操作。因为 \(9,10,4\) 使我们新建的节点,所以我们他们的 \(key\) 值是随机的,合并之后的树可能和原树不一样。这种情况下,\(4\) 就有可能成为 \(6\) 的右儿子,这就对其他版本的树产生了干扰,所以我们需要新建节点。

代码:


#include<bits/stdc++.h>
#define dd double
#define ld long double
#define ll long long
#define uint unsigned int
#define ull unsigned long long
#define N 30000010
#define M number
using namespace std;

const int INF=2147483647;

template<typename T> inline void read(T &x) {
    x=0; int f=1;
    char c=getchar();
    for(;!isdigit(c);c=getchar()) if(c == '-') f=-f;
    for(;isdigit(c);c=getchar()) x=x*10+c-'0';
    x*=f;
}

inline int random(int n){return 1ll*rand()*rand()%n+1;}

struct Node{
    int val,size,ls,rs,key;
    inline Node(){}
    inline Node(int val,int size,int ls,int rs,int key) : val(val),size(size),ls(ls),rs(rs),key(key) {}
    inline void Print(){
        printf("%d %d %d %d %d\n",val,size,ls,rs,key);
    }
};

Node p[N];
int tot,root[N],rt;

#define ls(k) p[k].ls
#define rs(k) p[k].rs
struct FHQ_Treap{
    inline int NewNode(int val){
        ++tot;p[tot]=Node(val,1,0,0,random(INF));return tot;
    }
    inline void PushUp(int k){p[k].size=p[ls(k)].size+p[rs(k)].size+1;}
    inline void Split(int k,int val,int &x,int &y){
        if(!k){x=y=0;return;}
        if(p[k].val<=val){x=k;Split(rs(k),val,rs(k),y);}
        else{y=k;Split(ls(k),val,x,ls(y));}PushUp(k);
    }
    inline int Merge(int x,int y){
        if(!x||!y) return x+y;
        if(p[x].key<=p[y].key){ls(y)=Merge(x,ls(y));PushUp(y);return y;}
        else{rs(x)=Merge(rs(x),y);PushUp(x);return x;}
    }
    inline void Split_(int k,int val,int &x,int &y){
        // printf("k=%d\n",k);printf("ls=%d rs=%d\n",ls(k),rs(k));
        if(!k){x=y=0;return;}
        if(p[k].val<=val){++tot;p[tot]=p[k];x=tot;Split_(rs(x),val,rs(x),y);PushUp(x);}
        else{++tot;p[tot]=p[k];y=tot;Split_(ls(y),val,x,ls(y));PushUp(y);}
    }
    inline int Merge_(int x,int y){
        if(!x||!y) return x+y;
        if(p[x].key<=p[y].key){int k=++tot;p[k]=p[y];ls(k)=Merge_(x,ls(k));PushUp(k);return k;}
        else{int k=++tot;p[k]=p[x];rs(k)=Merge_(rs(k),y);PushUp(k);return k;}
    }
    inline void Insert(int last,int val,int id){
        int x,y;Split_(root[last],val,x,y);root[id]=Merge_(Merge_(x,NewNode(val)),y);
    }
    inline void Delete(int last,int val,int id){
        int x,y,z;Split_(root[last],val-1,x,y);
        Split_(y,val,y,z);if(y) y=Merge_(ls(y),rs(y));
        root[id]=Merge_(Merge_(x,y),z);
    }
    inline int GetRank(int last,int val,int id){
        root[id]=root[last];int x,y;Split(root[id],val-1,x,y);int ans=p[x].size+1;
        root[id]=Merge(x,y);return ans;
    }
    inline int GetVal(int last,int rank,int id){
        root[id]=root[last];int k=root[id];
        // printf("root[%d]=%d\n",id,root[id]);
        // printf("ls=%d rs=%d val=%d\n",ls(k),rs(k),p[k].val);
        // printf("%d\n",p[1].val);
        while(k){
            if(p[ls(k)].size+1==rank) return p[k].val;
            else if(p[ls(k)].size+1>rank) k=ls(k);
            else{rank-=p[ls(k)].size+1;k=rs(k);}
        }return INF-3;
    }
    inline int GetNext(int last,int val,int id){
        root[id]=root[last];int x,y;Split(root[id],val,x,y);
        int k=y;while(ls(k)) k=ls(k);int ans;if(!k) ans=INF;else ans=p[k].val;
        root[id]=Merge(x,y);return ans;
    }
    inline int GetPre(int last,int val,int id){
        root[id]=root[last];int x,y;Split(root[id],val-1,x,y);
        int k=x;while(rs(k)) k=rs(k);int ans;if(!k) ans=-INF;else ans=p[k].val;
        root[id]=Merge(x,y);return ans;
    }
}tr;

int n;

int main(){
    // freopen("my.in","r",stdin);
    // freopen("my.out","w",stdout);
    read(n);
    for(int i=1;i<=n;i++){
        int last,op,val;
        read(last);read(op);read(val);
        if(op==1) tr.Insert(last,val,i);
        else if(op==2) tr.Delete(last,val,i);
        else if(op==3) printf("%d\n",tr.GetRank(last,val,i));
        else if(op==4) printf("%d\n",tr.GetVal(last,val,i));
        else if(op==5) printf("%d\n",tr.GetPre(last,val,i));
        else if(op==6) printf("%d\n",tr.GetNext(last,val,i));
    }
    return 0;
}

上一篇:阿里云 MaxCompute 2020-10 月刊


下一篇:阿里云机器学习PAI和大数据平台MaxCompute分别进入IDC和Forrester领导者象限