树状数组+二分答案查询第k大的数 (团体程序设计天梯赛 L3-002. 堆栈)

前提是数的范围较小

 1 数据范围:O(n) 
 2 查第k大的数i：log(n)(树状数组查询小于等于i的数目)*log(n)(二分找到i)
 3 添加：log(n) (树状数组)
 4 删除：log(n) (树状数组)

团体程序设计天梯赛 L3-002. 堆栈

/*数据范围:O(n) 
查第k大的数i：log(n)(树状数组查询小于等于i的数目)*log(n)(二分找到i)
添加：log(n) (树状数组)
删除：log(n) (树状数组)
*/ 
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <set>
#include <vector>
#include <stack>
#include <map>
#include <queue>
#include <algorithm>
#include <iostream>
using namespace std;
#define maxn 100000

long a[100005];
long st[100005];

int main()
{
    long n,d,l,r,mid,i,g,c=0;
    char s[20];
    for (i=0;i<=maxn;i++)
        a[i]=0;
    scanf("%ld",&n);
    while (n)
    {
        n--;
        scanf("%s",s);
        if (strcmp(s,"Pop")==0)
        {
            if (c==0)
            {
                printf("Invalid\n");
                continue;
            }    
            d=st[c];
            printf("%ld\n",d);
            while (d<=maxn)
            {
                a[d]--;
                d+=(d & (-d));
            }
            c--;
        }
        else if (strcmp(s,"Push")==0)
        {
            scanf("%ld",&d);
            c++;
            st[c]=d;            
            while (d<=maxn)
            {
                a[d]++;
                d+=(d & (-d));
            }
        }
        else
        {
            d=(c+1)>>1;
            if (c==0)
            {
                printf("Invalid\n");
                continue;                
            }
            l=1; r=maxn;
            while (l<=r)
            {
                mid=(l+r)>>1;
                g=0;
                i=mid;
                while (i>=1)
                {
                    g+=a[i];
                    i-=(i & (-i));
                }
                if (g>=d)
                    r=mid-1;
                else
                    l=mid+1;
            }
            printf("%ld\n",l);
        }
    }
    return 0;
}