PAT L3-002. 堆栈

树状数组，二分。

一堆数字，可以删除栈顶，压入数字，求中位数，可以线段树，也可以树状数组上二分。

#include<map>
#include<set>
#include<ctime>
#include<cmath>
#include<queue>
#include<string>
#include<stack>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
using namespace std;

stack<int>S;
int n; char op[];

int c[];

int lowbit(int x)
{
    return x&(-x);
}

int get(int p)
{
    int res=;
    while(p>)
    {
        res=res+c[p];
        p=p-lowbit(p);
    }
    return res;
}

void update(int p,int val)
{
    while(p<=)
    {
        c[p]=c[p]+val;
        p=p+lowbit(p);
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=;i<=n;i++)
    {
        scanf("%s",op);
        if(strcmp(op,"Pop")==)
        {
            if(S.empty()) printf("Invalid\n");
            else
            {
                printf("%d\n",S.top());
                update(S.top(),-); S.pop();
            }
        }
        else if(strcmp(op,"Push")==)
        {
            int x; scanf("%d",&x);
            S.push(x);
            update(x,);
        }
        else
        {
            if(S.size()==)
            {
                printf("Invalid\n");
                continue;
            }
            int limit = (S.size()+)/;
            int L=,R=,p=-;

            while(L<=R)
            {
                int mid = (L+R)/;
                if(get(mid)>=limit) p=mid,R=mid-;
                else L=mid+;
            }

            printf("%d\n",p);
        }
    }
    return ;
}