PAT L3-002： 堆栈（线段树）

https://www.patest.cn/contests/gplt/L3-002

题意：中文题意。

思路：因为所有数<=1e5，权值线段树维护每个数出现多少次，然后每次出栈入栈都更新权值就好了。

好久没写线段树了，写的好爽233。

#include <bits/stdc++.h>
using namespace std;
#define N 100010
#define lson rt<<1, l, m
#define rson rt<<1|1, m + 1, r
int tree[N<<2], val[N];
stack<int> sta;

void pushup(int rt) { tree[rt] = tree[rt<<1] + tree[rt<<1|1]; }

void update(int rt, int l, int r, int id, int val) {
    if(l == r && l == id) { tree[rt] += val; return ; }
    if(l == r) return ;
    int m = (l + r) >> 1;
    if(id <= m) update(lson, id, val);
    else update(rson, id, val);
    pushup(rt);
}

int query(int rt, int l, int r, int id) {
    if(l == r) return l;
    int m = (l + r) >> 1;
    if(tree[rt<<1] >= id) return query(lson, id);
    else return query(rson, id - tree[rt<<1]);
}

int main() {
    int n; scanf("%d", &n);
    char s[20]; int x, now; int ma = 100000;
    for(int i = 1; i <= n; i++) {
        scanf("%s", s);
        if(s[1] == 'o') {
            if(sta.empty()) { puts("Invalid"); continue; }
            now = sta.top(); sta.pop();
            update(1, 1, ma, now, -1);
            printf("%d
", now);
        } else if(s[1] == 'e') {
            if(sta.empty()) { puts("Invalid"); continue; }
            now = query(1, 1, ma, (sta.size() + 1) / 2);
            printf("%d
", now);
        } else {
            scanf("%d", &x);
            sta.push(x);
            update(1, 1, ma, x, 1);
        }
    }
    return 0;
}