插入输出(TREAP)

#include <stdlib.h>
#include

struct Node{

int size,key,weight;
Node* l,* r;
Node(int key):key(key){
	size=1;
	weight=rand();
	l=r=NULL;
}

};
int get_size(Nodenode){
return node==NULL?0:node->size;
}
void left_rotate(Node
&a){
Node* b=a->r;
a->r=b->l;
b->l=a;
b->size=a->size;
a->size=get_size(a->l)+get_size(a->r)+1;
a=b;
}
void right_rotate(Node*&a){
Node* b=a->l;
a->l=b->r;
b->r=a;
b->size=a->size;
a->size=get_size(a->l)+get_size(a->r)+1;
a=b;
}
void insert(Node*&node,int key){
if(!node){
node=new Node(key);
}else if(key<=node->key){
node->size++;
insert(node->l,key);
if(node->l->weightweight)right_rotate(node);
}
else{
node->size++;
insert(node->r,key);
if(node->r->weightweight)left_rotate(node);
}
}
int find_kth(Nodenode,int k){
int lsize=get_size(node->l);
if(k==lsize+1)return node->key;
else if(k<=lsize)return find_kth(node->l,k);
else return find_kth(node->r,k-1-lsize);
}
int A[30000];
int main(){
Node
root=NULL;
int m,n;
int now=1;
scanf("%d %d",&m,&n);
for(int i=1;i<=m;i++)scanf("%d",&A[i]);
int u;
for(int i=1;i<=n;i++){
scanf("%d",&u);
for(;now<=u;now++)
insert(root,A[now]);
printf("%d\n",find_kth(root,i));
}
return 0;
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值