草稿

来源:互联网 发布:南京 知乎 编辑:程序博客网 时间:2024/06/02 03:48
#include<stdio.h>
#include<string.h>
#define N 100005
int root,val[N],lch[N],rch[N],w[N],size;
void rrot(int &t) {
int x=lch[t]; lch[t]=rch[x]; rch[x]=t;
w[x]=w[t]; w[t]=w[lch[t]]+w[rch[t]]+1;
t=x;
}
void lrot(int &t) {
int x=rch[t]; rch[t]=lch[x]; lch[x]=t;
w[x]=w[t]; w[t]=w[lch[t]]+w[rch[t]]+1;
t=x;
}
void mant(int &t) {
int flag=0;
if(w[lch[lch[t]]]>w[rch[t]]) flag=1, rrot(t);
else if(w[rch[lch[t]]]>w[rch[t]]) flag=1, lrot(lch[t]), rrot(t);
else if(w[rch[rch[t]]]>w[lch[t]]) flag=1, lrot(t);
else if(w[lch[rch[t]]]>w[lch[t]]) flag=1, rrot(rch[t]), lrot(t);
if(flag) mant(lch[t]), mant(rch[t]), mant(t);
}
int select(int k) {
int t=root;
while(k!=w[lch[t]]+1) {
if(k<=w[t]) t=lch[t];
else k-=w[lch[t]]+1, t=rch[t];
}
return val[k];
}
int rank(int v) {
int t=root,ret=1;
while(v!=val[t]) {
if(v<val[t]) t=lch[t];
else ret+=w[lch[t]]+1, t=rch[t];
}
return ret;
}
void insert(int v) {
if(!root) {root=++size; val[size]=v; w[size]=1; return; }
int t=root,*x;
while(w[t]++) {
if(v<val[t]) {
if(lch[t]) t=lch[t];
else {x=&lch[t]; break; }
}
else {
if(rch[t]) t=rch[t];
else {x=&rch[t]; break; }
}
}
*x=++size; val[size]=v; w[size]=1;
mant(root);
}
void erase(int t,int v) {
int pre=0,*x;
while(w[t]--) {
if(v<val[t]) {pre=t; t=lch[t]; continue; }
if(v>val[t]) {pre=t; t=rch[t]; continue; }
x=&t; break;
}
if(!pre) {root=0; return; }
if(!lch[t]&&!rch[t]) {*x=0; return; }
if(!lch[t]) {*x=rch[t]; return; }
if(!rch[t]) {*x=lch[t]; return; }

}
void dfs(int u) {
printf("%d ",val[u]);
putchar('(');
if(lch[u]) dfs(lch[u]);
if(rch[u]) dfs(rch[u]);
putchar(')');
}
int main() {
insert(5);
insert(46);
insert(34632);
insert(87);
insert(876);
insert(123);
insert(7856);
dfs(root);
puts("");
printf("%d\n",select(3));
}
原创粉丝点击