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);
}
}
}