PTA团体程序设计天梯赛-练习集 L3-002 特殊堆栈（树状数组）

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

输入格式：

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

Push key
Pop
PeekMedian
其中Push表示入栈，key是不超过105的正整数；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
参考博客：https://blog.csdn.net/Yaokai_AssultMaster/article/details/79492190
https://www.liuchuo.net/archives/2263

#include <bits/stdc++.h>
using namespace std;
#define lowbit(i) ((i) & (-i))
/*
*lowbit(2) ->
*				i -> 0010
*				-i -> 1110
*lowbit(2) -> ((i) & (-i)) -> 0010
*i += lowbit(i) -> 0010 + 0010 = 0100
*/
const int maxn = 100010;
int c[maxn];
stack<int> s;
//更新树状数组
void update(int x,int v){
    for(int i = x;i < maxn;i += lowbit(i))
        c[i] += v;
}
//查询小于等于x的数的个数
int getsum(int x){
    int sum = 0;
    for(int i = x;i >= 1;i -= lowbit(i))
        sum += c[i];
    return sum;
}
void PeekMedian(){
	//二分查找
	//偶数或奇数都可以把k看成是(s.size() + 1) / 2;
    int left = 1,right = maxn - 1,mid,k = (s.size() + 1) / 2;
    while(left < right){
        mid = (left + right) / 2;
        if(getsum(mid) >= k)
            right = mid;
        else
            left = mid + 1;
    }
    cout<<left<<endl;
}
int main() {
    int n,num;
    cin>>n;
    for(int i = 0;i < n;i++){
        string str;
        cin>>str;
        if(str[1] == 'u'){
            cin>>num;
            s.push(num);
            update(num,1);
        }
        else if(str[1] == 'o'){
            if(!s.empty()){
                update(s.top(),-1);
                cout<<s.top()<<endl;
                s.pop();
            }
            else{
                cout<<"Invalid"<<endl;
            }
        }
        else{
            if(!s.empty()){
                PeekMedian();
            }
            else{
                cout<<"Invalid"<<endl;
            }
        }
    }
    return 0;
}