hdu 2852 树状数组

思路：加一个数e就用update(e,1)。删除元素e就用update(e,-1)。找比a大的第k大的元素就用二分查找。

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

#define Maxn 120010

#define lowbit(x) (x&(-x))

using namespace std;

int C[Maxn];

int Sum(int pos)

{

    int sum=0;

    while(pos)

    {

        sum+=C[pos];

        pos-=lowbit(pos);

    }

    return sum;

}

void update(int pos,int val)

{

    while(pos<=120010)

    {

        C[pos]+=val;

        pos+=lowbit(pos);

    }

}

int main()

{

    int m,i,j,p,k,e,num,a;

    while(scanf("%d",&m)!=EOF)

    {

        num=0;

        memset(C,0,sizeof(C));

        while(m--)

        {

            scanf("%d",&p);

            if(p==0)

            {

                scanf("%d",&e);

                update(e,1);

                num++;

            }

            if(p==1)

            {

                scanf("%d",&e);

                if(Sum(e)-Sum(e-1))

                {

                    update(e,-1);

                    num--;

                }

                else

                    printf("No Elment!\n");

            }

            if(p==2)

            {

                scanf("%d%d",&a,&k);

                int l,r,mid,temp;

                int mins;

                mins=Sum(a);

                l=a,r=120000;

                if(!num||num-mins<k)

                {

                    printf("Not Find!\n");

                    continue;

                }

                while(l<r)

                {

                    mid=(l+r)>>1;

                    temp=Sum(mid)-mins;

                    if(temp>=k)

                        r=mid;

                    else

                        l=mid+1;

                }

                if(r>a)

                    printf("%d\n",r);

                else

                    printf("Not Find!\n");

            }

        }

    }

    return 0;

}