天梯赛－L3-002(树状数组)

L3-002 特殊堆栈 (30分)

堆栈是一种经典的后进先出的线性结构，相关的操作主要有“入栈”（在堆栈顶插入一个元素）和“出栈”（将栈顶元素返回并从堆栈中删除）。本题要求你实现另一个附加的操作：“取中值”——即返回所有堆栈中元素键值的中值。给定 N 个元素，如果 N 是偶数，则中值定义为第 N/2 小元；若是奇数，则为第 (N+1)/2 小元。

---

输入格式：
输入的第一行是正整数N≤10^5。随后 N 行，每行给出一句指令，为以下 3 种之一：
Push key
Pop
PeekMedian
其中 key 是不超过 10
​5
​​ 的正整数；Push 表示“入栈”；Pop 表示“出栈”；PeekMedian 表示“取中值”。

---

输出格式：
对每个 Push 操作，将 key 插入堆栈，无需输出；对每个 Pop 或 PeekMedian 操作，在一行中输出相应的返回值。若操作非法，则对应输出 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

---

思路其实简单的说1e5的数据没有办法直接暴力的跑暴力这跑必会时间超限，所以我们被迫换个思路，就是用树状数组记录他每个点维护的数值，然后二分查询结果让求得个数居中就行。

---

/**

 _  ___             _______
| |/ (_)           |___  / |
| ' / _ _ __   __ _   / /| |__   __ _ _ __   __ _
|  < | | '_ \ / _` | / / | '_ \ / _` | '_ \ / _` |
| . \| | | | | (_| |/ /__| | | | (_| | | | | (_| |
|_|\_\_|_| |_|\__, /_____|_| |_|\__,_|_| |_|\__, |
               __/ |                         __/ |
              |___/                         |___/

**/
#include <stack>
#include <map>
#include <queue>
#include <string>
#include<iostream>
#include<stdio.h>
#include<string.h>
#include <algorithm>
#include <math.h>
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
typedef pair<ll, ll> pii;
#define mem(a,x) memset(a,x,sizeof(a))
#define debug(x) cout << #x << ": " << x << endl;
#define rep(i,n) for(int i=0;i<(n);++i)
#define repi(i,a,b) for(int i=int(a);i<=(b);++i)
#define repr(i,b,a) for(int i=int(b);i>=(a);--i)
const int maxn = 1e5 + 10;
#define inf 0x3f3f3f3f
#define sf scanf
#define pf printf
stack<int> s;
string str;
int sum[maxn]={0};
int lowbit(int x){
	return (x & (-x));
}
void add(int x,int y){
	for(int i=x;i<maxn;i+=lowbit(i)){
		sum[i]+=y;
	}
}
int getsum(int x){
	int num=0;
	for(int i=x;i>=1;i-=lowbit(i)){
		num+=sum[i];
	}
	return num;
}
void ack(){
	int l=1,r=maxn,k=(s.size()+1)/2;
	while(l<r){
		int mid=(l+r)/2;
		if(getsum(mid)>=k){
			r=mid;
		}else l=mid+1;
	}
	pf("%d\n",r);
}
int main(){
	int n;
	cin>>n;
	while(n--){
		cin>>str;
		if(str=="Pop"){
			if(s.empty()){
				puts("Invalid");
			}else {
				add(s.top(),-1);
				pf("%d\n",s.top());
				s.pop();
			}
		}
		else if(str=="Push"){
			int m;
			sf("%d",&m);
			s.push(m);
			add(m,1);
		}else if(str=="PeekMedian"){
			if(s.empty()){
				puts("Invalid");
			}else ack();
		}
	}
	return 0;
}