团体程序设计天梯赛-练习集 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 <bits/stdc++.h>
using namespace std;
int tree[101000<<2];
stack<int> S;

void update(int l, int r, int rt, int v, int add)
{
    if(l == r)
    {
        tree[rt] += add;
        return ;
    }
    int mid = (l + r) / 2;
    if(v <= mid) update(l, mid, rt*2, v, add);
    else update(mid + 1, r, rt*2+1, v, add);
    tree[rt] = tree[rt*2] + tree[rt*2+1];
}

int find(int l,int r, int rt, int v)
{
    if(l == r) return l;
    int mid = (l + r) / 2;
    if(tree[rt*2] >= v) return find(l, mid, rt*2, v);
    else return find(mid+1, r, rt*2+1, v - tree[rt*2]);
}

int main()
{
    int T, t, sz;
    char ch[100];
    scanf("%d", &T);
    while(T--)
    {
        scanf("%s", ch);
        if(!strcmp(ch, "Pop"))
        {
            if(S.empty()) printf("Invalid\n");
            else
            {
                t = S.top();
                S.pop();
                printf("%d\n", t);
                update(0, 100000, 1, t, -1);
            }
        }
        else if(!strcmp(ch,"PeekMedian"))
        {
            if(S.empty())
            {
                printf("Invalid\n");
                continue;
            }
            sz = S.size();
            if(sz % 2) sz = (sz+1) / 2;
            else sz = sz / 2;
            printf("%d\n", find(0, 100000, 1, sz));
        }
        else if(!strcmp(ch,"Push"))
        {
            scanf("%d", &t);
            S.push(t);
            update(0, 100000, 1, t, 1);
        }
    }
    return 0;
}