1. 程式人生 > 實用技巧 >非旋Treap

非旋Treap

Treap是一種動態樹,其主要操作是旋轉rotate,不過我比較喜歡splay(附上鍊接可以看看哦kkkkk),而這樣就不能支援持久化了

為了強行優化這一點,我們決定將其轉化為靜態的Treap,每插入一個新的資料,都先複製上一個版本,再進行修改,這樣會極大的

節省空間。我們採取使用隨機權值fix來強行修改樹的結構,降低樹的深度,變成了log,這樣,就在持久化的同時進行了優化

#include<iostream>
#include<cstdio>
#include<string>
#include<algorithm>
#include<string>
#define lc(x) ch[x][0]
#define rc(x) ch[x][1]
#define N 300010
#define M (N*50)

using namespace std;

typedef pair <int ,int > dat;
int ch[M][2],val[M],siz[M],fix[M];//val維護BST的性質,fix維護斜堆的性質 
int cnt=0,n,rt[N];
+
inline int Read() { int num=0,k=1; char c=getchar(); while(c!='-'&&(c<'0'||c>'9')) c=getchar(); if(c=='-'){k=-1;c=getchar();} while(c>='0'&&c<='9'){num=(num<<1)+(num<<3)+(c^48);c=getchar();} return num*k; } inline void Copy (int a,int b) { val[a]=val[b];siz[a]=siz[b]; ch[a][0]=ch[b][0];ch[a][1]=ch[b][1]; fix[a]=fix[b]; } inline void push_up(int o) { siz[o]=siz[lc(o)]+siz[rc(o)]+1; } inline int New_node (int x=0) +
{ val[++cnt]=x; siz[cnt]=1; ch[cnt][0]=ch[cnt][1]=0; fix[cnt]=rand()%5000; return cnt; } inline int merge(int a,int b)//利用fix強行維護堆的性質,判斷依據是其fix值 { //但實際上我們同時維護了BST的性質,這就是為什麼合併到左右子樹會有區別 if(!a || !b) return a+b; int x=++cnt; if(fix[a]<fix[b]) { Copy(x,a); ch[x][1]=merge(ch[x][1],b); } else { Copy(x,b); ch[x][0]=merge(a,ch[x][0]); } push_up(x); return x; } inline dat split(int o,int k)//分裂 { if(!o) return make_pair(0,0); int x=++cnt; Copy(x,o); dat t; if(k<=siz[lc(x)]) { t=split(lc(x),k); ch[x][0]=t.second; push_up(x); return make_pair(t.first,x); } else { t=split(rc(x),k-siz[lc(x)]-1); ch[x][1]=t.first; push_up(x); return make_pair(x,t.second); } } int insert(int o,int k,int x) { dat t=split(o,k-1); int tmp=New_node(x); int ret=merge(merge(t.first,tmp),t.second); return ret; } int del(int o,int k) { dat a=split(o,k-1); dat b=split(a.second,1); return merge(a.first,b.second); } int No_k(int o,int k) { if(!o)return 0; if(k==siz[lc(o)]+1) return val[o]; if(k<=siz[lc(o)])return No_k(lc(o),k); return No_k(rc(o),k-siz[lc(o)]-1); } void dfs(int o) { if(!o) return; dfs(lc(o)); printf("%d ",val[o]); dfs(rc(o)); } int main() { int tot=0; rt[0]=0; n=Read(); for(int i=1;i<=n;i++) { int op,t,k,x; op=Read();t=Read();k=Read(); if(op==1) x=Read(); switch(op) { case 1:{rt[++tot]=insert(rt[t],k,x);break;} case 2:{rt[++tot]=del(rt[t],k);break;} case 3: {printf("%d\n",No_k(rt[t],k));break;} } } return 0; }