板子题
SBT:
1 #include<cstdio> 2 struct SBT { 3 int v, sz, ch[2], cnt; 4 } t[300005]; 5 int n, cnt, root; 6 #define Upd(k) { 7 t[k].sz = t[t[k].ch[0]].sz + t[t[k].ch[1]].sz + t[k].cnt; 8 } 9 void rot(int &k, bool f) { 10 int p = t[k].ch[f]; t[k].ch[f] = t[p].ch[!f]; t[p].ch[!f] = k; 11 Upd(k); Upd(p); k = p; 12 } 13 inline void mt(int &k,bool f) 14 { 15 if(!k) return; 16 if(t[t[k].ch[f^1]].sz < t[t[t[k].ch[f]].ch[f]].sz) rot(k, f); 17 else if(t[t[k].ch[f^1]].sz < t[t[t[k].ch[f]].ch[f^1]].sz) { rot(t[k].ch[f], f^1); rot(k, f); } 18 else return; 19 mt(t[k].ch[f],f); 20 mt(k,f); 21 } 22 void Ins(int &k, int x) { 23 if(!k) {k = ++ cnt; t[k].sz = t[k].cnt = 1; t[k].v = x; return; } 24 ++ t[k].sz; 25 if(t[k].v == x) {++ t[k].cnt; return;} 26 Ins(t[k].ch[t[k].v < x], x); 27 mt(k, t[k].v < x); 28 } 29 int Del(int &k, int x) { 30 if(!k) return k; 31 int tmp; 32 if(t[k].v == x) { 33 if(t[k].cnt > 1) {-- t[k].cnt; --t[k].sz; return k;} 34 else if(!(t[k].ch[0]*t[k].ch[1])) {k = t[k].ch[0]+t[k].ch[1];} 35 else { tmp = Del(t[k].ch[0], x+1); t[k].cnt = t[tmp].cnt; t[k].v = t[tmp].v; Upd(k); return k; } 36 } 37 else if((t[k].v < x && !t[k].ch[1]) || (t[k].v > x && ! t[k].ch[0])) { tmp = k; k = t[k].ch[0]; Upd(k); return tmp; } 38 else tmp = Del(t[k].ch[t[k].v < x], x); 39 Upd(k); return tmp; 40 } 41 int Ran(int k,int x) { 42 if(k==0)return 0; 43 if(t[k].v==x)return t[t[k].ch[0]].sz+1; 44 else if(x>t[k].v) 45 return t[t[k].ch[0]].sz+t[k].cnt+Ran(t[k].ch[1],x); 46 else return Ran(t[k].ch[0],x); 47 } 48 int Kth(int k,int x) { 49 if(k==0)return 0; 50 if(x<=t[t[k].ch[0]].sz) 51 return Kth(t[k].ch[0],x); 52 else if(x>t[t[k].ch[0]].sz+t[k].cnt) 53 return Kth(t[k].ch[1],x-t[t[k].ch[0]].sz-t[k].cnt); 54 else return t[k].v; 55 } 56 int ans; 57 void pred(int k,int x) { 58 if(k==0)return; 59 if(t[k].v<x) { 60 ans=k; 61 pred(t[k].ch[1],x); 62 } else pred(t[k].ch[0],x); 63 } 64 void succ(int k,int x) { 65 if(k==0)return; 66 if(t[k].v>x) { 67 ans=k; 68 succ(t[k].ch[0],x); 69 } else succ(t[k].ch[1],x); 70 } 71 char c, f; 72 inline void GET(int &n) { 73 n = 0; f = 1; 74 do {c = getchar(); if(c == '-') f = -1;} while(c > '9' || c < '0'); 75 while(c >= '0' && c <= '9') {n=n*10+c-'0';c=getchar();} 76 n *= f; 77 } 78 int main() { 79 GET(n); 80 int opt,x; 81 for(int i=1; i<=n; i++) { 82 GET(opt); GET(x); 83 switch(opt) { 84 case 1: Ins(root,x); break; 85 case 2: Del(root,x); break; 86 case 3: printf("%d ",Ran(root,x)); break; 87 case 4: printf("%d ",Kth(root,x)); break; 88 case 5: ans=0; pred(root,x); printf("%d ",t[ans].v); break; 89 case 6: ans=0; succ(root,x); printf("%d ",t[ans].v); break; 90 } 91 } 92 return 0; 93 }
Treap:
1 #include<iostream> 2 #include<cstdio> 3 #include<cstdlib> 4 using namespace std; 5 struct data { 6 int ch[2], v, sz, rnd, cnt; 7 } tr[300005]; 8 int n,size,root,ans,sd = 2333; 9 inline int ran() { return sd = (sd*sd+sd)%(10003); } 10 inline void Upd(int k) { 11 tr[k].sz = tr[tr[k].ch[0]].sz + tr[tr[k].ch[1]].sz + tr[k].cnt; 12 } 13 void rot(int &k, bool f) { ///0->right_rotate 14 int t = tr[k].ch[f]; tr[k].ch[f] = tr[t].ch[!f]; tr[t].ch[!f] = k; 15 Upd(k); Upd(t); k = t; 16 } 17 void insert(int &k,int x) { 18 if(k==0) { size++; k=size; tr[k].sz=tr[k].cnt=1; tr[k].v=x; tr[k].rnd=ran(); return; } 19 tr[k].sz++; 20 if(tr[k].v==x) { tr[k].cnt++; return; } 21 bool f = x>tr[k].v; 22 insert(tr[k].ch[f],x); 23 if(tr[tr[k].ch[f]].rnd<tr[k].rnd)rot(k, f); 24 } 25 26 void del(int &k,int x) { 27 if(k==0)return; 28 if(tr[k].v==x) { 29 if(tr[k].cnt>1) { tr[k].cnt--; tr[k].sz--; return; } 30 if(tr[k].ch[0]*tr[k].ch[1]==0)k=tr[k].ch[0]+tr[k].ch[1]; 31 else rot(k, tr[tr[k].ch[0]].rnd >= tr[tr[k].ch[1]].rnd), del(k, x); 32 } else tr[k].sz--, del(tr[k].ch[x>tr[k].v], x); 33 } 34 int get_rank(int k,int x) { 35 if(k==0)return 0; 36 if(tr[k].v==x)return tr[tr[k].ch[0]].sz+1; 37 else if(x>tr[k].v) 38 return tr[tr[k].ch[0]].sz+tr[k].cnt+get_rank(tr[k].ch[1],x); 39 else return get_rank(tr[k].ch[0],x); 40 } 41 int get_kth(int k,int x) { 42 if(k==0)return 0; 43 if(x<=tr[tr[k].ch[0]].sz) 44 return get_kth(tr[k].ch[0],x); 45 else if(x>tr[tr[k].ch[0]].sz+tr[k].cnt) 46 return get_kth(tr[k].ch[1],x-tr[tr[k].ch[0]].sz-tr[k].cnt); 47 else return tr[k].v; 48 } 49 void pred(int k,int x) { 50 if(k==0)return; 51 if(tr[k].v<x) { 52 ans=k; 53 pred(tr[k].ch[1],x); 54 } else pred(tr[k].ch[0],x); 55 } 56 void succ(int k,int x) { 57 if(k==0)return; 58 if(tr[k].v>x) { 59 ans=k; 60 succ(tr[k].ch[0],x); 61 } else succ(tr[k].ch[1],x); 62 }char c, f; 63 inline void GET(int &n) { 64 n = 0; f = 1; 65 do {c = getchar(); if(c == '-') f = -1;} while(c > '9' || c < '0'); 66 while(c >= '0' && c <= '9') {n=n*10+c-'0';c=getchar();} 67 n *= f; 68 } 69 int main() { 70 GET(n); 71 int opt,x; 72 for(int i=1; i<=n; i++) { 73 GET(opt); GET(x); 74 switch(opt) { 75 case 1: insert(root,x); break; 76 case 2: del(root,x); break; 77 case 3: printf("%d ",get_rank(root,x)); break; 78 case 4: printf("%d ",get_kth(root,x)); break; 79 case 5: ans=0; pred(root,x); printf("%d ",tr[ans].v); break; 80 case 6: ans=0; succ(root,x); printf("%d ",tr[ans].v); break; 81 } 82 } 83 return 0; 84 }
Splay
1 #include<cstdio> 2 #define MAXN 300005 3 struct node { 4 int v, cnt, sz, ch[2], f; 5 }t[MAXN]; 6 int rt, sz, n; 7 #define Upd(r) {t[r].sz = t[t[r].ch[0]].sz + t[t[r].ch[1]].sz + t[r].cnt;} 8 void rot(int x) 9 { 10 int y = t[x].f, z = t[y].f; 11 bool f = (t[y].ch[1] == x); 12 t[y].ch[f] = t[x].ch[f^1]; 13 if(t[y].ch[f]) t[t[y].ch[f]].f = y; 14 t[x].ch[f^1] = y; t[y].f = x; 15 t[x].f = z; 16 if(z) t[z].ch[t[z].ch[1]==y] = x; 17 Upd(y); 18 } 19 void Spaly(int r, int tp) { 20 for(int y, z; (y = t[r].f) != tp; rot(r)) { 21 z = t[y].f; 22 if(z == tp) continue; 23 if( (t[z].ch[0] == y) == (t[y].ch[0] == r) ) rot(y); 24 else rot(r); 25 } 26 if(!tp) rt = r; Upd(r); 27 } 28 void Ins(int r, int x) { 29 int y = 0; 30 while(r && t[r].v != x) { y = r; r = t[r].ch[x > t[r].v]; } 31 if(r) ++ t[r].cnt; 32 else { 33 r = ++ sz; t[r].sz = t[r].cnt = 1; 34 t[r].v = x; t[r].f = y; if(y) t[y].ch[x > t[y].v] = r; 35 } 36 Spaly(r, 0); 37 } 38 void Find(int v) { 39 int x = rt; if(!x) return; 40 while(t[x].ch[v > t[x].v] && t[x].v != v) x = t[x].ch[v > t[x].v]; 41 Spaly(x, 0); 42 } 43 int Ran(int v) { 44 Find(v); 45 return t[t[rt].ch[0]].sz; 46 } 47 int Kth(int x) 48 { 49 int y=rt,p; 50 if(x>t[rt].sz)return 0; 51 while(1) 52 { 53 p=t[y].ch[0]; 54 if(t[p].sz+t[y].cnt<x) { 55 x-=t[p].sz+t[y].cnt; 56 y=t[y].ch[1]; 57 } 58 else if(t[p].sz>=x) y=p; 59 else return t[y].v; 60 } 61 } 62 int Nxt(int x, bool f) 63 { 64 Find(x); 65 if((t[rt].v>x&&f)||(t[rt].v<x&&!f)) return rt; 66 int p = t[rt].ch[f]; 67 while(t[p].ch[f^1]) p = t[p].ch[!f]; 68 return p; 69 } 70 void Del(int v) { 71 int p = Nxt(v, 0), s = Nxt(v, 1); 72 Spaly(p, 0); Spaly(s, p); 73 p = t[s].ch[0]; 74 if(t[p].cnt > 1) -- t[p].cnt, Spaly(p, 0); 75 else t[s].ch[0] = 0; 76 } 77 char c, f; 78 inline void GET(int &n) { 79 n = 0; f = 1; 80 do {c = getchar(); if(c == '-') f = -1;} while(c > '9' || c < '0'); 81 while(c >= '0' && c <= '9') {n=n*10+c-'0';c=getchar();} 82 n *= f; 83 } 84 int main() { 85 GET(n); 86 int opt,x; 87 Ins(rt, -0x7fffffff); Ins(rt, +0x7fffffff); 88 for(int i=1; i<=n; i++) { 89 GET(opt); GET(x); 90 switch(opt) { 91 case 1: Ins(rt,x); break; 92 case 2: Del(x); break; 93 case 3: printf("%d ",Ran(x)); break; 94 case 4: printf("%d ",Kth(x+1)); break; 95 case 5: printf("%d ",t[Nxt(x, 0)].v); break; 96 case 6: printf("%d ",t[Nxt(x, 1)].v); break; 97 } 98 } 99 return 0; 100 }