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;
}