Dynamic Rankings（zoj 2112）

题意：带修改的第K大

#include<cstdio>
#include<iostream>
#include<cstring>
#define N 400010
#define inf 1000000000
using namespace std;
int a[N],ans[N],sum[N],tmp[N],n,m,num,cnt;
struct node{
    int x,y,s,k,tp,cur;
};node q[N],q1[N],q2[N];
void modify(int x,int v){
    while(x<=n){
        sum[x]+=v;
        x+=x&(-x);
    }
}
int query(int x){
    int tot=;
    while(x){
        tot+=sum[x];
        x-=x&(-x);
    }
    return tot;
}
void solve(int head,int tail,int l,int r){
    if(head>tail)return;
    if(l==r){
        for(int i=head;i<=tail;i++)
            if(q[i].tp==)ans[q[i].s]=l;
        return;
    }
    int mid=l+r>>;
    for(int i=head;i<=tail;i++){
        if(q[i].tp==&&q[i].y<=mid)modify(q[i].x,);
        if(q[i].tp==&&q[i].y<=mid)modify(q[i].x,-);
        if(q[i].tp==) tmp[i]=query(q[i].y)-query(q[i].x-);
    }
    for(int i=head;i<=tail;i++){
        if(q[i].tp==&&q[i].y<=mid)modify(q[i].x,-);
        if(q[i].tp==&&q[i].y<=mid)modify(q[i].x,);
    }
    int l1=,l2=;
    for(int i=head;i<=tail;i++){
        if(q[i].tp==){
            if(q[i].cur+tmp[i]>=q[i].k) q1[++l1]=q[i];
            else q[i].cur+=tmp[i],q2[++l2]=q[i];
        }
        else {
            if(q[i].y<=mid) q1[++l1]=q[i];
            else q2[++l2]=q[i];
        }
    }
    for(int i=;i<=l1;i++) q[head+i-]=q1[i];
    for(int i=;i<=l2;i++) q[head+l1+i-]=q2[i];
    solve(head,head+l1-,l,mid);
    solve(head+l1,tail,mid+,r);
}
void work(){
    for(int i=;i<=n;i++){
        scanf("%d",&a[i]);
        q[++num].x=i;q[num].y=a[i];q[num].tp=;q[num].s=;
    }
    scanf("%d",&m);
    for(int i=;i<=m;i++){
        int op;scanf("%d",&op);
        if(op==){
            int x,y;scanf("%d%d",&x,&y);
            q[++num].x=x;q[num].y=a[x];q[num].tp=;q[num].s=;
            q[++num].x=x;q[num].y=y;q[num].tp=;q[num].s=;
            a[x]=y;
        }
        else {
            int x,y,k;scanf("%d%d%d",&x,&y,&k);
            q[++num].x=x;q[num].y=y;q[num].k=k;q[num].tp=;q[num].s=++cnt;
        }
    }
    solve(,num,-inf,inf);
    for(int i=;i<=cnt;i++)
        printf("%d\n",ans[i]);
}
int main(){
    while(scanf("%d",&n)!=EOF){
        memset(ans,,sizeof(ans));
        memset(tmp,,sizeof(tmp));
        memset(sum,,sizeof(sum));
        memset(q,,sizeof(q));
        num=cnt=;
        work();
    }
    return ;
}