HDU2852 KiKi's K-Number 线段树单点更新

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2852

题目大意：给定一个容器有3中操作：“0 x”表示插入元素x；“1 x”表示删除元素x（如果x不存在输出”No Elment!”）；“2 x k”表示查询比x大的数中，第k个元素的值（同理，不存在该值输出“Not Find!”）。给你m个操作（m小于10^6），分别模拟每种操作。

分析：很明显，单纯的模拟操作肯定会TLE，那么我们可以用线段树sum来纪录每一区间内元素的总个数，然后对于插入和删除操作，我们分别令该元素所在的区间的sum值加减1即可；对于查询x是否存在，我们可以分别查询[1,x]和[1,x-1]所在区间的sum值，如果不相同，说明x是存在的；找出第k个比x大的元素操作，我们可以把这个操作分为两步，先找出容器内比x小的数有多少个t，即表示出x在容器中是第t大的数，然后我们只需找出容器中第t+k大的数即可。

实现代码如下：

#include <cstdio>
using namespace std;
const int maxn=1000005;
struct segment
{
    int l,r;
    int sum;//纪录区间内有多少个数
}tree[maxn<<2];
void build(int root,int l,int r)
{
    tree[root].l=l;
    tree[root].r=r;
    tree[root].sum=0;
    if(l==r) return ;
    int mid=(l+r)>>1;
    build(root<<1,l,mid);
    build(root<<1|1,mid+1,r);
}
void update(int root,int v,int w)
{
    if(tree[root].l==v&&tree[root].r==v)
    {
        tree[root].sum+=w;
        return ;
    }
    int mid=(tree[root].l+tree[root].r)>>1;
    if(v<=mid) update(root<<1,v,w);
    else update(root<<1|1,v,w);
    tree[root].sum=tree[root<<1].sum+tree[root<<1|1].sum;
}
int query(int root,int l,int r)
{
    if(l>tree[root].r||r<tree[root].l)
      return 0;
    if(l<=tree[root].l&&r>=tree[root].r)
      return tree[root].sum;
    int mid=(tree[root].l+tree[root].r)>>1;
    int sum1=0,sum2=0;
    if(l<=mid) sum1=query(root<<1,l,r);
    if(r>mid) sum2=query(root<<1|1,l,r);
    return sum1+sum2;
}
int ans;
void solve(int root,int k)
{
    int mid=(tree[root].l+tree[root].r)>>1;
    if(k>tree[root<<1].sum)
    {
        if(tree[root].l==tree[root].r)
        {
            ans=tree[root].r;
            return ;
        }
        solve(root<<1|1,k-tree[root<<1].sum);
    }
    else solve(root<<1,k);
}
int main()
{
    int m,p,v,k;
    while(scanf("%d",&m)!=-1)
    {
        build(1,1,maxn);
        while(m--)
        {
            scanf("%d",&p);
            if(p==0)
            {
                scanf("%d",&v);
                update(1,v,1);
            }
            else if(p==1)
            {
                scanf("%d",&v);
                if(query(1,1,v)==query(1,1,v-1))
                  puts("No Elment!");
                else update(1,v,-1);
            }
            else
            {
                scanf("%d%d",&v,&k);
                int t=query(1,1,v);
                //printf("%d\n",t);
                solve(1,t+k);
                if(ans==maxn) puts("Not Find!");
                else printf("%d\n",ans);
            }
        }
    }
    return 0;
}