草稿

#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));
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值