线段树+L3-002. 堆栈

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

using namespace std;

queue<int > q;
stack<int > s;
int n,node[100005*4],p;

void update(int l,int r,int n,int p,int flag)
{
    if(l==r&&p==l)
    {
       if(flag)
        node[n]++;
       else
        node[n]--;
        return ;
    }
    int mid=l+(r-l)/2;
    if(p<=mid)
      update(l,mid,2*n,p,flag);
    else
        update(mid+1,r,2*n+1,p,flag);
    node[n]=node[n*2]+node[n*2+1];
}

void find(int l,int r,int n,int p)
{
    if(l==r&&p<=node[n])
    {
        printf("%d\n",l);
        return ;
    }
    int m=l+(r-l)/2;
    if(node[n*2]>=p)
        find(l,m,2*n,p);
    else
        find(m+1,r,n*2+1,p-node[n*2]);
}

int main()
{
    memset(node,0,sizeof(node));
    scanf("%d",&n);
    while(n--)
    {
        char str[25];
        scanf("%s",str);
        if(str[1]=='o')
        {
            if(s.empty())
                printf("Invalid\n");
            else
                {
                    p=s.top();
                    printf("%d\n",p);
                    s.pop();
                    update(1,100010,1,p,0);
                }
        }
        else if(str[1]=='u')
        {
            scanf("%d",&p);
            s.push(p);
            update(1,100010,1,p,1);
        }
        else if(str[1]=='e')
        {
            if(!s.empty())
            {
            p=(s.size()+1)/2;
            find(1,100010,1,p);
            }
            else
                printf("Invalid\n");
        }
    }
    return 0;
}