板子题
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\n",Ran(root,x)); break;
87 case 4: printf("%d\n",Kth(root,x)); break;
88 case 5: ans=0; pred(root,x); printf("%d\n",t[ans].v); break;
89 case 6: ans=0; succ(root,x); printf("%d\n",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\n",get_rank(root,x)); break;
78 case 4: printf("%d\n",get_kth(root,x)); break;
79 case 5: ans=0; pred(root,x); printf("%d\n",tr[ans].v); break;
80 case 6: ans=0; succ(root,x); printf("%d\n",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\n",Ran(x)); break;
94 case 4: printf("%d\n",Kth(x+1)); break;
95 case 5: printf("%d\n",t[Nxt(x, 0)].v); break;
96 case 6: printf("%d\n",t[Nxt(x, 1)].v); break;
97 }
98 }
99 return 0;
100 }
时间： 2024-10-10 12:01:52