1057 Stack （30 分）

题目介绍

题意：有三个操作,push,pop,peek。peek代表求栈内元素的中值
输入的元素<=1e5 操作n<=1e5
思路：用树状数组动态更新、a[x]表示权值为x的数量。
那么查询的时候sum(x)就代表了所有<=x的数量，这样从1~N进行二分查找
找到sum(x)== (len+1)/2 那么x就是中值了

代码

#include<bits/stdc++.h>
using namespace std;

#define ll long long
#define pb push_back
#define mp make_pair

const int N = 1e5+4;

int c[N];

void update(int x,int val){
   for(int i=x;i<N;i+= (i &-i))
       c[i]+=val;
}
int sum(int x){
   int res= 0;
   for(int i=x;i;i-=(i&-i))res+=c[i];
   return res;
}
int get(int x){
   int l =0;  int r= N;
   int ans;
   while(l<=r){
       int mid =(l+r)>>1;
       if(sum(mid) >=x){
           ans = mid;
           r = mid-1;
       }
       else l=mid+1;
   }
   return ans;
}


int main(){

   int n;
   cin>>n;
   char op[222];

   stack<int>S;

   while(n--){
       scanf("%s",op);
       //pop
       if(op[1]=='o'){
           if(S.size()){
               printf("%d\n",S.top());
               update(S.top(),-1);
               S.pop();
           }
           else {
               printf("Invalid\n");
           }
       }
       else if(op[1]=='e'){
           if(S.size()){
               int len  =S.size();
               printf("%d\n",get( (len+1)/2 )  );
           }
           else   printf("Invalid\n");
       }
       //push
       else if(op[1]=='u'){
           int x;
           scanf("%d",&x);
           update(x,1);
           S.push(x);
       }

   }

   return 0;
}