HDOJ 2852 KiKi's K-Number(樹狀數組)

超級傳送門：http://acm.hdu.edu.cn/showproblem.php?pid=2852

題意：有一個容器裏有m個數（1<=m<100000），可以加入新的數，也可以刪掉已有的數。有多次查詢，查詢容器中第k大的數（非遞減序列的第k個元素）。

分析：用樹狀數組來維護這個容器，每次查找用二分。

代碼：

#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 100010;
int c[maxn];
int n;
int lowbit(int t) {
    return t & -t;
}
int getsum(int end) {
    int ret=0;
    for (int i=end;i>=1;i-=lowbit(i)) {
        ret += c[i];
    }
    return ret;
}
void update(int p,int val) {
    for (int i=p;i<maxn;i+=lowbit(i)) {
        c[i] += val;
    }
}
void query(int a,int k) {
    int l = a+1;
    int r = maxn-1;
    int res,mid,ans=-1;
    k += getsum(a);
    while (l<=r) {
        mid = (l+r)/2;
        res = getsum(mid);
        if (res>=k) {
            r = mid-1;
            ans = mid;
        } else {
            l = mid+1;
        }
    }
    if (ans==-1) printf("Not Find!\n");
    else printf("%d\n",ans);
    return;
}
int main () {
    #ifndef ONLINE_JUDGE
        freopen("1.txt","r",stdin);
    #endif
    int cmd,e,a,k;
    while (scanf("%d",&n)>0) {
        memset(c,0,sizeof(c));
        for (int i=1;i<=n;i++) {
            scanf("%d",&cmd);
            if (cmd==0) {
                scanf("%d",&e);
                update(e,1);
            } else if (cmd==1) {
                scanf("%d",&e);
                if (getsum(e)-getsum(e-1)==0) printf("No Elment!\n");
                else update(e,-1);
            } else if (cmd==2) {
                scanf("%d%d",&a,&k);
                query(a,k);
            }
        }
    }
    return 0;
}