PAT 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

用一个线段树来维护某段区间上所有数的个数，刚开始以为要用到主席树，后来看到数值的取值只有1e5，直接用线段树来维护这个大区间就可以了。详见代码

#include<iostream>
#include<bits/stdc++.h>
using namespace std;
int tree[1123450];
stack<int>A;

void update(int left,int right,int rt,int value,int add)
{
  //  cout<<rt<<endl;
    if(left==right)
    {
        tree[rt]+=add;
        return ;
    }
    int mid=(left+right)>>1;
    if(value<=mid) update(left,mid,rt<<1,value,add);
    else update(mid+1,right,rt<<1|1,value,add);
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}

int qu(int left,int right,int rt,int value)
{
    if(left==right) return left;
    int mid=(left+right)>>1;
    if(tree[rt<<1]>=value) return qu(left,mid,rt<<1,value);
    else return qu(mid+1,right,rt<<1|1,value-tree[rt<<1]);
}

int main()
{
    int n; cin>>n;
    memset(tree,0,sizeof(tree));
    char ch[66];
    while(!A.empty()) A.pop();
    while(n--)
    {
        scanf("%s",&ch);
        if(!strcmp(ch,"Pop"))
        {
            if(A.empty()) printf("Invalid\n");
            else
            {
                int tt=A.top();
                A.pop();
                printf("%d\n",tt);
                update(0,100000,1,tt,-1);
            }
        }
        else if(!strcmp(ch,"PeekMedian"))
        {
            if(A.empty())
            {
                printf("Invalid\n");
                continue;
            }
            int sz=A.size();
            if(sz%2) sz=(sz+1)/2;
            else sz=sz/2;
            printf("%d\n",qu(0,100000,1,sz));
        }
        else if(!strcmp(ch,"Push"))
        {
            int x; scanf("%d",&x);
            A.push(x);
            update(0,100000,1,x,1);
        }
    }
    return 0;
}