1057 Stack (30 分)

#include<cstdio>
#include<stack>
#include<string>
#include<iostream>
const int maxn=100010;
using namespace std;
#define lowbit(i) ((i)&(-i))
int c[maxn]={0};
stack<int> sta;
void update(int n,int x){
    for(int i=n;i<maxn;i+=lowbit(i)){
        c[i]+=x;
    }
}
int getSum(int n){
    int sum=0;
    for(int i=n;i>0;i-=lowbit(i)){
        sum+=c[i];
    }
    return sum;
}
int func(int k){
    int l=1,r=maxn-1;
    while(l<r){
        int mid=(l+r)/2;
        int num=getSum(mid);
        if(num>=k){
            r=mid;
        }
        else if(num<k){
            l=mid+1;
        }
    }
    return l;
}
int main(){
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        string s;
        cin>>s;
        if(s=="Pop"){
            if(sta.empty()){
                printf("Invalid\n");
                continue;
            }
            else {
                update(sta.top(),-1);
                printf("%d\n",sta.top());
                sta.pop();
            }
        }
        else if(s=="Push"){
            int x;
            scanf("%d",&x);
            update(x,1);
            sta.push(x);
        }
        else if(s=="PeekMedian"){
            if(sta.empty()){
                printf("Invalid\n");
                continue;
            }
            int a=func((sta.size()+1)/2);
            printf("%d\n",a);
        }
    }
    return 0;
}

总结：

如果用stack+hash+二分的方法话，hash每次计算前面有几个数字时，需要O(n)时间复杂度，找到所要的位置需要O(n*logn)，会超时，利用树状数组替代的话可以使计算时间缩短为O(logn)，总查找复杂度优化为O(logn^2)。