树状数组+二分答案查询第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[];
 long st[];

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