HDU 2852 KiKi's K-Number

权值线段树

#include <cstdio>
#include <cstring>
const int N=200000,M=220000;
int k,q,x,y,sum[M<<2],n;
char c;
void up(int x){sum[x]=sum[x<<1]+sum[x<<1|1];}
void update(int t,int num,int l,int r,int x){
    if(l==r){
        if(sum[x]+num<0)puts("No Elment!");
        else sum[x]+=num;return;
    }
    int mid=(l+r)>>1;
    if(t<=mid)update(t,num,l,mid,x<<1);
    else update(t,num,mid+1,r,x<<1|1);
    up(x);
}
int query(int L,int R,int l,int r,int x){
    if(L<=l&&r<=R)return(sum[x]);
    int mid=(l+r)>>1,tmp=0;
    if(L<=mid)tmp+=query(L,R,l,mid,x<<1);
    if(mid<R)tmp+=query(L,R,mid+1,r,x<<1|1);
    return tmp;
}
int find(int k,int l,int r,int x){
    int mid=(l+r)>>1,tmp=sum[x<<1|1];
    if(l==r)return l;
    if (k<=tmp) return find(k,mid+1,r,x<<1|1);
    else return find(k-tmp,l,mid,x<<1);
}
int scan(int &x){
    while(c=getchar(),c<'0'||c>'9');x=c-'0';
    while(c=getchar(),c>='0'&&c<='9')x=x*10+c-'0';
}
int main(){
    while(~scanf("%d",&n)){
        for(int i=0;i<(M<<2);i++)sum[i]=0;
        while(n--){
            scan(q);
            if(q==0){scan(x),update(x,1,1,N,1);}
            if(q==1){scan(x),update(x,-1,1,N,1);}
            if(q==2){
                scan(x),scan(k);
                if((y=query(x+1,N,1,N,1))<k)puts("Not Find!");
                else printf("%d\n",find(y-k+1,1,N,1));
            }
        }
    }
    return 0;
}

分块

#include <cstdio>
int i,j,x,y,k,op,m,n,p[1000],pos[100005],a[100005],block,st[100005],en[100005];
int main(){
    int size=0;
    while(size*size<=100000)size++;
    for(int i=1;i<=100000;i++)pos[i]=(i-1)/size+1;
    for(block=pos[100000],i=1;i<=block;i++)en[i-1]=(st[i]=size*(i-1)+1)-1;en[block]=100000;
    while(~scanf("%d",&n)){
        for(int i=1;i<1000;i++)p[i]=0;
        for(int i=1;i<=100000;i++)a[i]=0;
        while(n--){
            scanf("%d",&op);
            if(op==0){scanf("%d",&x);a[x]++;p[pos[x]]++;}
            if(op==1){scanf("%d",&x);if(!a[x])puts("No Elment!");else{a[x]--;p[pos[x]]--;}}
            if(op==2){
                scanf("%d%d",&x,&k);
                for(i=x+1;i<=en[pos[x+1]];i++){k-=a[i];if(k<=0){printf("%d\n",i);break;}}
                if(k>0){
                    for(i=pos[x+1]+1;i<=block;i++){k-=p[i];if(k<=0)break;}
                    if(k>0)puts("Not Find!");
                    else for(k+=p[i],j=st[i];j<=en[i];j++){k-=a[j];if(k<=0){printf("%d\n",j);break;}}
                }
            }
        }
    }return 0;
}