团体程序设计天梯赛L3-002 堆栈

L3-002. 堆栈

时间限制

200 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

using namespace std;

#define LL long long

const int INF=0x3f3f3f3f;

int a[100050],n,sum[100050*20];
char ch[20];

void f(int k,int l,int r,int val,int p)
{
    sum[k]+=p;
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(val<=mid) f(k<<1,l,mid,val,p);
    else f(k<<1|1,mid+1,r,val,p);
}

int Find(int k,int l,int r,int p)
{
    if(l==r) return l;
    int mid=(l+r)>>1;
    if(p>sum[k<<1]) return Find(k<<1|1,mid+1,r,p-sum[k<<1]);
    else return Find(k<<1,l,mid,p);
}

int main()
{
    while(~scanf("%d",&n))
    {
        memset(sum,0,sizeof sum);
        int cnt=0;
        while(n--)
        {
            scanf("%s",ch);
            if(!strcmp(ch,"Pop"))
            {
                if(!cnt) printf("Invalid\n");
                else
                {
                    f(1,1,100000,a[cnt-1],-1);
                    printf("%d\n",a[--cnt]);
                }
            }
            else if(!strcmp(ch,"Push"))
            {
                scanf("%d",&a[cnt++]);
                f(1,1,100000,a[cnt-1],1);
            }
            else
            {
                if(!cnt) printf("Invalid\n");
                else
                {
                    int k;
                if(cnt%2==1) k=(cnt+1)/2;
                else k=cnt/2;
                printf("%d\n",Find(1,1,100000,k));
                }
            }
        }
    }
    return 0;
}