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

题目描述

给出一个长度为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 做法，这次发一发 Spaly 做法。。。

思路相同，都是 kth。

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#define MAXN 100010
using namespace std;
int n,root=0,c=1;
struct node{
	int son[2];
	int f,v,s,flag;
}a[MAXN];
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;
}
inline void pushup(int rt){
	if(!rt)return;
	a[rt].s=a[rt].flag;
	if(a[rt].son[0])a[rt].s+=a[a[rt].son[0]].s;
	if(a[rt].son[1])a[rt].s+=a[a[rt].son[1]].s;
}
inline void turn(int rt,int k){
	int x=a[rt].f,y=a[x].f;
	a[x].son[!k]=a[rt].son[k];
	if(a[rt].son[k])a[a[rt].son[k]].f=x;
	a[rt].f=y;
	if(y)a[y].son[a[y].son[1]==x]=rt;
	a[x].f=rt;
	a[rt].son[k]=x;
	pushup(x);pushup(rt);
}
void splay(int rt,int ancestry){
	while(a[rt].f!=ancestry){
		int x=a[rt].f,y=a[x].f;
		if(y==ancestry)turn(rt,a[x].son[0]==rt);
		else{
			int k=a[y].son[0]==x?1:0;
			if(a[x].son[k]==rt){turn(rt,!k);turn(rt,k);}
			else{turn(x,k);turn(rt,k);}
		}
	}
	if(ancestry==0)root=rt;
}
void insert(int rt,int x){
	int fa=0;
	while(rt&&x!=a[rt].v){
		fa=rt;
		rt=a[rt].son[a[rt].v<x];
	}
	if(rt)a[rt].flag++;
	else{
		rt=c++;
		if(fa)a[fa].son[a[fa].v<x]=rt;
		a[rt].son[0]=a[rt].son[1]=0;
		a[rt].f=fa;a[rt].v=x;
		a[rt].s=a[rt].flag=1;
	}
	splay(rt,0);
}
int kth(int rt,int x){
	if(x>a[rt].s)return 0;
	while(1){
		int y=a[rt].son[0];
		if(x>a[y].s+a[rt].flag){
			x-=a[y].s+a[rt].flag;
			rt=a[rt].son[1];
		}
		else if(a[y].s>=x)rt=y;
		else return a[rt].v;
	}
}
int main(){
	n=read();
	for(int i=1;i<=n;i++){
		int x=read();
		insert(root,x);
		if(i%2)printf("%d\n",kth(root,i/2+1));
	}
	return 0;
}