堆栈

大家都知道“堆栈”是一种“先进后出”的线性结构，基本操作有“入栈”（将新元素插入栈顶）和“出栈”（将栈顶元素的值返回并从堆栈中将其删除）。现请你实现一种特殊的堆栈，它多了一种操作叫“查中值”，即返回堆栈中所有元素的中值。对于N个元素，若N是偶数，则中值定义为第N/2个最小元；若N是奇数，则中值定义为第(N+1)/2个最小元。

输入格式：

输入第一行给出正整数N（<= 10⁵）。随后N行，每行给出一个操作指令，为下列3种指令之一：

Push key
Pop
PeekMedian

其中Push表示入栈，key是不超过10⁵的正整数；Pop表示出栈；PeekMedian表示查中值。

输出格式：

对每个入栈指令，将key入栈，并不输出任何信息。对每个出栈或查中值的指令，在一行中打印相应的返回结果。若指令非法，就打印“Invalid”。

输入样例：

17
Pop
PeekMedian
Push 3
PeekMedian
Push 2
PeekMedian
Push 1
PeekMedian
Pop
Pop
Push 5
Push 4
PeekMedian
Pop
Pop
Pop
Pop

输出样例：

Invalid
Invalid
3
2
2
1
2
4
4
5
3
Invalid

#include<stdio.h>
#include<string.h>
#include<stack>
using namespace std;
const int N = 1123450;
int tree[N];
void update(int root,int left,int right,int val,int add){
	if(left==right){
		tree[root]+=add;
		return;
	}
	int mid=(left+right)/2;
	if(val<=mid) update(root*2,left,mid,val,add);
	else update(root*2+1,mid+1,right,val,add);
	tree[root]=tree[root*2]+tree[root*2+1];
}
int query(int root,int left,int right,int val){
	if(left==right) return left;
	int mid=(left+right)/2;
	if(tree[root*2]>=val) return query(root*2,left,mid,val);
	else return query(root*2+1,mid+1,right,val-tree[root*2]);
}
int main(){
	int T;
	scanf("%d",&T);
	stack<int>p;
	while(T--){
		char op[15];
		scanf("%s",op);
		if(!strcmp(op,"Pop")){
			if(p.empty()) printf("Invalid\n");
			else{
				int x=p.top();
				p.pop();
				printf("%d\n",x);
				update(1,1,100000,x,-1);
			}
		}
		else if(!strcmp(op,"PeekMedian")){
			if(p.empty()) printf("Invalid\n");
            else{
            	int sz=p.size();
            	if(sz%2) sz=(sz+1)/2;
            	else sz=sz/2;
            	printf("%d\n",query(1,1,100000,sz));
			}
		}
		else if(!strcmp(op,"Push")){
			int x;
			scanf("%d",&x);
			p.push(x);
			update(1,1,100000,x,1);
		}
	}
	return 0;
}