关于使用平衡树动态求解中位数

最新推荐文章于 2024-07-14 18:15:43 发布

东臻的记录

最新推荐文章于 2024-07-14 18:15:43 发布

阅读量362

点赞数 7

文章标签：算法

本文链接：https://blog.csdn.net/jiangdongzhen/article/details/136840708

版权

本文介绍了如何利用C++中的Treap（自平衡二叉搜索树）数据结构实现动态求解中位数问题，包括插入操作和查询给定位置的中位数节点值。

摘要由CSDN通过智能技术生成

洛谷直达

关于使用平衡树动态求解中位数

~~当然stl set<int>也可以~~

#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <string.h>
using namespace std;
const int N=2e4;
struct node 
{
	node *lf=NULL,*rt=NULL;
	const int rawk=rand();
	int date,siz=1,cnt=1;
	node(const int x){date=x;}
};
struct treap
{
	protected:
	node *root=NULL;
	inline void pu(node *cur){
		if(cur==NULL){return;}
		int temp=cur->cnt;
		temp+=cur->lf==NULL? 0: cur->lf->siz;
		temp+=cur->rt==NULL? 0: cur->rt->siz;
		cur->siz=temp;
	}
	void zig(node *&cur){
		if(cur==NULL){return;}
		node *temp=cur->lf;
		cur->lf=temp->rt;
		temp->rt=cur;
		pu(cur),pu(temp);
		cur=temp;
	}
	void zag(node *&cur){
		if(cur==NULL){return;}
		node *temp=cur->rt;
		cur->rt=temp->lf;
		temp->lf=cur;
		pu(cur),pu(temp);
		cur=temp;
	}
	void insert_sx(node *&cur,int x){
		if(cur==NULL){cur=new node(x);return;}
		if(cur->date>x){
			insert_sx(cur->lf,x);
			if(cur->rawk>cur->lf->rawk){zig(cur);}
		}
		else if(cur->date<x){
			insert_sx(cur->rt,x);
			if(cur->rawk>cur->rt->rawk){zag(cur);}
		}
		else{++cur->cnt;}
		pu(cur);
	}
	node *rankval_sx(node *cur,int rk){
		int temp=cur->lf==NULL? 0:cur->lf->siz;
		if(rk<=temp){return rankval_sx(cur->lf,rk);}
		if(rk<=temp+cur->cnt){return cur;}
		return rankval_sx(cur->rt,rk-temp-cur->cnt);
	}
	public:
	void insert(int x){insert_sx(root,x);}
	node *rankval(int rk){return rankval_sx(root,rk);}
}t;
int n,G,now;
char cz[10];
int main()
{
	scanf("%d",&now);G=now;
	--now;
	while(G--)
	{
		scanf("%d",&n);
		t.insert(n);
	}
	scanf("%d",&G);
	while(G--)
	{
		scanf("%s",cz);
		if(strcmp(cz,"add")==0)
		{scanf("%d",&n);t.insert(n);now++;}
		else
		{printf("%d\n",t.rankval((now/2)+1)->date);}
	}
	
	

    return 0;
}