PAT甲1057 Stack（30 分）

#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <string>
#include <stack>
using namespace std;
#define lowbit(i) ((i)&(-i))
const int maxn=100010;
int C[100010],N,num;


void update(int x,int v)
{
    for(int i=x;i<maxn;i+=lowbit(i))
    {
        C[i]+=v;
    }
}

int getsum(int x)
{
    int sum=0;
    for(int i=x;i>0;i-=lowbit(i))
    {
        sum+=C[i];
    }
    return sum;
}

int getmid(int num)
{
    int k=(num+1)/2;
    int l=1,r=maxn-1;
    while(l<r)
    {
        int mid=(l+r)/2;
        if(getsum(mid)>=k)
        {
            r=mid;
        }
        else
        {
            l=mid+1;
        }
    }
    return l;
}


int main()
{
    scanf("%d",&N);
    char op[20];
    stack<int> s;
    for(int i=0;i<N;i++)
    {
        scanf("%s",&op);
        if(strcmp(op,"Pop")==0)
        {
            if(s.size()!=0)
            {
                int top=s.top();
                s.pop();
                update(top,-1);
                printf("%d\n",top);
            }
            else
            {
                printf("Invalid\n");
            }
        }
        else if(strcmp(op,"Push")==0)
        {
            int a;
            scanf("%d",&a);
            s.push(a);
            update(a,1);
        }
        else if(strcmp(op,"PeekMedian")==0)
        {
            if(s.size()!=0)
            {
                  int a=getmid(s.size());
                    printf("%d\n",a);
            }
            else
            {
                printf("Invalid\n");
            }
        }
    }
    return 0;
}