FHQ-TREAP && 可持久化平衡树

FHQ—TREAP
只要学过splay,线段树合并分裂,就非常好学吧。。。。
持久化的话,就是在合并分裂的时候,新建一个节点来保存,以免对先前节点造成影响。。。。

FHQ-TREAP(持久化)

#include<bits/stdc++.h>
#define MAXN 10000005
#define maxn 600005
typedef long long ll;
using namespace std;

int n,tot,rt[maxn];
ll mod;

struct node{
	int ch[2],val,sz,rnd;
}t[MAXN];

void up(int rt){
	t[rt].sz = 1;
	if(t[rt].ch[0])t[rt].sz += t[t[rt].ch[0]].sz;
	if(t[rt].ch[1])t[rt].sz += t[t[rt].ch[1]].sz;
}

int mem(int x , int y){
	if(!x||!y)return (x + y);
	if(t[x].rnd < t[y].rnd){
		t[x].ch[1] = mem(t[x].ch[1] , y);
		up(x);
		return x;
	}
	else{
		t[y].ch[0] = mem(x , t[y].ch[0]);
		up(y);
		return y;
	}
}

void split(int now , int &x , int &y , int k){
	if(!now){
		x = y = 0;
		return;
	}
	if(t[now].val <= k){
		x = now;
		split(t[now].ch[1] , t[x].ch[1] , y , k);
	}
	else{
		y = now;
		split(t[now].ch[0] , x , t[y].ch[0] , k);
	}
	up(now);
}

int new_node(int v){
	tot++;
	t[tot].val = v;
	t[tot].sz = 1;
	return tot;
}

int copy(int now){
	if(!now)return 0;
	int zz = new_node(t[now].val);
	t[zz].ch[0] = copy(t[now].ch[0]);	
	t[zz].ch[1] = copy(t[now].ch[1]);
	up(zz);
	return zz;
}

int kth(int u , int k){
	if(t[t[u].ch[0]].sz < k){
		k -= t[t[u].ch[0]].sz;
		if(k == 1)return t[u].val;
		return kth(t[u].ch[1] , k - 1);
	}
	return kth(t[u].ch[0] , k);
}

ll nxt(int u){
	if(!u)return -mod + 1;
	while(t[u].ch[1])u = t[u].ch[1];
	return t[u].val;
}

ll pre(int u){
	if(!u)return mod - 1;
	while(t[u].ch[0])u = t[u].ch[0];
	return t[u].val;
}

int main(){
	mod = 1;
	for(int i = 1 ; i <= 31 ; i++)mod = (mod << 1);
	cin>>n;int v,tp,u,x,y,z,p;
	for(int i = 1 ; i <= n ; i++){
		cin>>v>>tp>>u;
		rt[i] = rt[v];
		if(tp == 1){
			x = y = 0;
			split(rt[i] , x , y , u - 1);
			rt[i] = mem(mem(x , new_node(u)) , y);	
		}
		if(tp == 2){
			x = y = z = p = 0;
			split(rt[i] , x , y , u);
			split(x , z , p , u - 1);
			p = mem(t[p].ch[0] , t[p].ch[1]);
			rt[i] = mem(mem(z , p) , y);			
		}
		if(tp == 3){
			x = y = 0;
			split(rt[i] , x , y , u - 1);
			cout<<t[x].sz + 1<<endl;
			rt[i] = mem(x , y);
		}
		if(tp == 4){
			cout<<kth(rt[i] , u)<<endl;
		}
		if(tp == 5){
			x = y = 0;
			split(rt[i] , x , y , u - 1);
			cout<<nxt(x)<<endl;
			rt[i] = mem(x , y);
		}
		if(tp == 6){
			x = y = 0;
			split(rt[i] , x , y , u);
			cout<<pre(y)<<endl;
			rt[i] = mem(x , y);
		}
	}
}

FHQ-THREP(UN持久化)

#include<bits/stdc++.h>
#define MAXN 300005
using namespace std;

int tot,RT,n;

struct node{
	int ch[2],sz,val,rnd;
}t[MAXN];

int new_node(int x){
	tot++;
	t[tot].val = x;
	t[tot].rnd = rand();
	t[tot].sz = 1;
	return tot;
}

void up(int rt){
	t[rt].sz = 1;
	if(t[rt].ch[0])t[rt].sz += t[t[rt].ch[0]].sz;
	if(t[rt].ch[1])t[rt].sz += t[t[rt].ch[1]].sz;
}

int mem(int x , int y){
	if(!x||!y)return (x + y);
	if(t[x].rnd < t[y].rnd){
		t[x].ch[1] = mem(t[x].ch[1] , y);
		up(x);
		return x;
	}	
	else{
		t[y].ch[0] = mem(x , t[y].ch[0]);
		up(y);
		return y;
	}
}

void split(int now , int &x , int &y , int k){//
	if(!now){
		x = y = 0;
		return;
	}
	if(t[now].val <= k){
		x = now;
		split(t[now].ch[1] , t[x].ch[1] , y , k);
	}
	else{
		y = now;
		split(t[now].ch[0] , x , t[y].ch[0] , k);
	}
	up(now);
}

void insert(int v){
	int x,y;
	x = y = 0;
	split(RT , x , y , v);
	RT = mem(mem(x , new_node(v)) , y);
}

void del(int v){
	int x , y , z , p;
	x = y = z = p = 0;
	split(RT , x , y , v);
	split(x , z , p , v - 1);
	p = mem(t[p].ch[0] , t[p].ch[1]);
	RT = mem(mem(z , p) , y);
}

int kth(int u , int k){
	if(t[t[u].ch[0]].sz < k){
		k -= t[t[u].ch[0]].sz;
		if(k == 1)return t[u].val;
		return kth(t[u].ch[1] , k - 1);
	}
	return kth(t[u].ch[0] , k);
}

int pre(int u){
	if(t[u].ch[0])return pre(t[u].ch[0]);
	return t[u].val;
}

int nxt(int u){
	if(t[u].ch[1])return nxt(t[u].ch[1]);
	return t[u].val;
}

int main(){
	cin>>n;int tp,x;
	while(n--){
		cin>>tp>>x;
		if(tp == 1)insert(x);	
		if(tp == 2)del(x);	
		if(tp == 3){
			int zz1 , zz2;zz1 = zz2 = 0;
			split(RT , zz1 , zz2 , x - 1);
			cout<<t[zz1].sz + 1<<endl;
			RT = mem(zz1 , zz2);
		}
		if(tp == 4)cout<<kth(RT , x)<<endl;
		if(tp == 5){
			int zz1 , zz2;zz1 = zz2 = 0;
			split(RT , zz1 , zz2 , x - 1);
			cout<<nxt(zz1)<<endl;
			RT = mem(zz1 , zz2);
		}
		if(tp == 6){
			int zz1 , zz2;zz1 = zz2 = 0;
			split(RT , zz1 , zz2 , x);
			cout<<pre(zz2)<<endl;
			RT = mem(zz1 , zz2);
		}
	}
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值