天梯赛-堆栈（线段树）

题目 点击打开链接

大家都知道“堆栈”是一种“先进后出”的线性结构，基本操作有“入栈”（将新元素插入栈顶）和“出栈”（将栈顶元素的值返回并从堆栈中将其删除）。现请你实现一种特殊的堆栈，它多了一种操作叫“查中值”，即返回堆栈中所有元素的中值。对于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

题目分析：实现一种能查找中值的栈结构。区间第K大问题。。

#include<iostream>
#include<stdio.h>
using namespace std;
int A[300000]={0};
void update(int root,int l,int r,int k,int x){
    if(l==r){
        if(x) A[root]++;
        else A[root]--;
    }
    else{
        int mid=(l+r)/2;
        if(k<=mid) update(root*2,l,mid,k,x);
        else update(root*2+1,mid+1,r,k,x);
        A[root]=A[root*2]+A[root*2+1];
    }

}
int q(int root,int l,int r,int k){
    if(r==l) return l;
    else{
        int mid=(l+r)/2;
        if(k<=A[root*2]) return q(root*2,l,mid,k);
        else return q(root*2+1,mid+1,r,k-A[root*2]);
    }
}
int main(){
    int B[100010],n,i,k=0;
    char s[1000];
    scanf("%d",&n);
    for(i=1;i<=n;i++){
        scanf("%s",&s);
        if(s[1]=='e'){
            if(k==0) printf("Invalid\n");
            else printf("%d\n",q(1,1,100010,(k+1)/2));
        }
        if(s[1]=='o'){
            if(k==0) printf("Invalid\n");
            else {
                printf("%d\n",B[k]);
                update(1,1,100010,B[k--],0);
            }
        }
        if(s[1]=='u'){
                scanf("%d",&B[++k]);
                update(1,1,100010,B[k],1);
        }
    }
    return 0;
}