hdu 2852 线段树 单点更新

主要是查询的时候 本来时间就比较紧 查询的时候直接暴力二分求答案 超时

看了下别人代码 主要是find的时候要直接find速度快了很多

#include<cstdio>
#include<cstring>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
const int N=100000;
using namespace std;
int sum[N<<3],ans,n,op,l,r,m,k,mid;
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void update(int p,int add,int l,int r,int rt)
{
    if(l==r){
        sum[rt]+=add;
        return;
    }
    int mid=(l+r)/2;
    if(p<=mid) update(p,add,lson);
    else update(p,add,rson);
    pushup(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R) return sum[rt];
    int mid=(l+r)/2,tmp=0;
    if(L<=mid) tmp+=query(L,R,lson);
    if(R>mid) tmp+=query(L,R,rson);
    return tmp;
}
int find_rt(int t,int l,int r,int rt)
{
    int mid=(l+r)/2;
    if(t>sum[rt<<1]){
        if(l==r)return r;
        return find_rt(t-sum[rt<<1],rson);
    }
    return find_rt(t,lson);
}
int main()
{//freopen("C:\\Users\\Administrator\\Desktop\\input.txt","r",stdin);
    while(scanf("%d",&n)!=EOF)
    {
        memset(sum,0,sizeof(sum));
        while(n--){
            scanf("%d",&op);
            if(op==0){scanf("%d",&m);update(m,1,1,N,1);}
            else if(op==1)
            {
                scanf("%d",&m);
                if(query(m,m,1,N,1)==0)
                    printf("No Elment!\n");
                else update(m,-1,1,N,1);
            }
            else if(op==2)
            {
                scanf("%d%d",&m,&k);
                int t=query(1,m,1,N,1)+k;
                ans=find_rt(t,1,N,1);
                if(ans==N) printf("Not Find!\n");
                else printf("%d\n",ans);
            }
        }
    }
    return 0;
}