洛谷P1168 中位数——动态中位数_Treap

题目描述

给出一个长度为N的非负整数序列A[i]，对于所有1 ≤ k ≤ (N + 1) / 2，输出A[1], A[3], …, A[2k - 1]的中位数。即前1，3，5，……个数的中位数。

输入输出格式

输入格式：

输入文件median.in的第1行为一个正整数N，表示了序列长度。

第2行包含N个非负整数A[i] (A[i] ≤ 10^9)。

输出格式：

输出文件median.out包含(N + 1) / 2行，第i行为A[1], A[3], …, A[2i – 1]的中位数。

输入输出样例

输入样例#1： 

7
1 3 5 7 9 11 6

输出样例#1： 

1
3
5
6

说明

对于20%的数据，N ≤ 100；

对于40%的数据，N ≤ 3000；

对于100%的数据，N ≤ 100000。

不想用 堆，于是打了发 Treap 的 kth，轻松多了。。。

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
int n;
struct node{
	node* son[2];
	int w,v,s,flag;
	node(){
		son[0]=son[1]=NULL;
		w=rand();
		v=0;
		s=flag=1;
	}
};
node* rt;
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
void maintain(node* &u){
	u->s=u->flag;
	if(u->son[0]!=NULL)u->s+=u->son[0]->s;
	if(u->son[1]!=NULL)u->s+=u->son[1]->s;
}
void turn(node* &u,int k){
	node* t=u->son[k^1];
	u->son[k^1]=t->son[k];
	t->son[k]=u;
	maintain(u);
	maintain(t);
	u=t;
}
void insert(node* &u,int x){
	if(u==NULL){
		u=new node;
		u->v=x;
		return;
	}
	else if(u->v==x){
		u->flag++;
		maintain(u);
		return;
	}
	int y=x>u->v?1:0;
	insert(u->son[y],x);
	if(u->son[y]->w>u->w)turn(u,y^1);
	else maintain(u);
}
int kth(node* u,int k){
	if(u==NULL)return 0;
	int lsons=0;
	if(u->son[0]!=NULL)lsons=u->son[0]->s;
	if(k>=lsons+1&&k<=u->flag+lsons)return u->v;
	if(k<=lsons)return kth(u->son[0],k);
	else return kth(u->son[1],k-u->flag-lsons);
}
int main(){
	n=read();
	srand(987);
	for(int i=1;i<=n;i++){
		int x=read();
		insert(rt,x);
		if(i%2)printf("%d\n",kth(rt,i/2+1));
	}
	return 0;
}