L3-002. 堆栈 -STL + 线段树单节点更新

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

/**
题意：
为栈赋予新功能，取第k个值；本题要求k=(n+1)/2
题解：
思路一:巧用STL
用stack判断操作是否合法
使用upper_bound(),lower_bound()函数快速找到需要做改动的值
然后使用vector.erase(),vector.insert()函数进行插入删除。

思路二：线段树
维护这样一棵树：每个节点存储元素个数，找第k个元素时，
如果当前节点lchild.cnt >= k，则说明在左儿子中，递归寻找左儿子
反之 说明在右儿子的第k-lchild.cnt个
*/


///******************************
///****         STL         *****
///******************************
#include<bits/stdc++.h>

using namespace std;
const int INF=0x3f3f3f3f;
stack<int>s;

int main(){
    vector <int> v;
    stack <int> s;
    vector <int> :: iterator it;
    int n,t;
    char a[20];
    scanf("%d",&n);
    for(int i = 0;i<n;++i){
        scanf("%s",a);
        if(!strcmp(a,"Pop")){
            if(s.empty()) puts("Invalid");
            else {
                int x = s.top();
                s.pop();
                printf("%d\n",x);
                v.erase(lower_bound(v.begin(),v.end(),x));
            }
        }else if(!strcmp(a,"Push")){

            scanf("%d",&t);
            s.push(t);
            ///注意insert函数是插在当前位置的前面
            v.insert(upper_bound(v.begin(),v.end(),t),t);
        }else{
            if(s.empty())puts("Invalid");
            else{
                int pos = (s.size()+1)/2-1;
                printf("%d\n",v[pos]);
            }
        }
    }
    return 0;
}


///******************************
///****       线段树        *****
///******************************
#include<bits/stdc++.h>

using namespace std;
const int INF=0x3f3f3f3f;
const int maxn = 100000+10;
stack<int>s;
struct node{
    int l,r,cnt;
}tree[maxn*4];

void build(int node, int l, int r){

    tree[node].l = l;
    tree[node].r = r;
    tree[node].cnt = 0;
    if(l==r) return;
    int mid = (l+r)/2;
    build(node*2,l,mid);
    build(node*2+1,mid+1,r);
}

void update(int node, int ql, int qr,int val){
    if(tree[node].l>=ql && tree[node].r<=qr){
        tree[node].cnt+=val;
        return;
    }
    int mid = (tree[node].l+tree[node].r)/2;
    if(ql<=mid) update(node*2,ql,qr,val);
    if(qr>mid) update(node*2+1,ql,qr,val);
    tree[node].cnt = tree[node*2].cnt + tree[node*2+1].cnt;
}

int query(int node, int q){
    if(tree[node].l == tree[node].r) return tree[node].l;
    if(tree[node*2].cnt>=q) query(node*2,q);
    else query(node*2+1,q-tree[node*2].cnt);
}

int main(){
    int n,t;
    char a[20];
    build(1,1,maxn);
    scanf("%d",&n);
    for(int i = 0;i<n;++i){
        scanf("%s",a);
        if(!strcmp(a,"Pop")){
            if(s.empty()) puts("Invalid");
            else {
                int x = s.top();
                s.pop();
                printf("%d\n",x);
                update(1,x,x,-1);
            }
        }else if(!strcmp(a,"Push")){
            scanf("%d",&t);
            s.push(t);
            update(1,t,t,1);
        }else{
            if(s.empty())puts("Invalid");
            else{
                int pos = (s.size()+1)/2;
                printf("%d\n",query(1,pos));
            }
        }
    }

    return 0;
}