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<<], val[N];
 stack<int> sta;

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

 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) >> ;
     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) >> ;
     if(tree[rt<<] >= id) return query(lson, id);
     else return query(rson, id - tree[rt<<]);
 }

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