treap增加操作查询操作

最新推荐文章于 2024-12-29 17:41:14 发布

oshawott_cute

最新推荐文章于 2024-12-29 17:41:14 发布

阅读量145

点赞数

CC 4.0 BY-SA版权

分类专栏： acm

本文链接：https://blog.youkuaiyun.com/qq_36172410/article/details/89508440

acm 专栏收录该内容

75 篇文章

订阅专栏

本文介绍了一种使用二叉搜索树进行元素排名查询和根据排名查询元素的方法。通过递归遍历树结构，可以高效地获取指定值的排名位置，以及根据排名找到对应的元素值。这种方法适用于需要频繁进行排名相关操作的场景。

//递归到叶子节点，一路维护信息即可
//查询排名

int query_rank(int k,int x){
	if (k==0) return 0;
	if (tr[k].v==x) return tr[tr[k].l].size+1;
	else if (x>tr[k].v)
	   return tr[tr[k].l].size+tr[k].w+query_rank(tr[k].r,x);
	else return query_rank(tr[k].l,x);
}

//查询排名对应的个数

int query_num(int k,int x){
	if (k==0) return 0;
	if (x<=tr[tr[k].l].size)
	   return query_num(tr[k].l,x);
	else if (x>tr[tr[k].l].size+tr[k].w)
	  return query_num(tr[k].r,x-tr[tr[k].l].size-tr[k].w);
	else return tr[k].v;
}