P2617 Dynamic Rankings(整体二分)

动态修改的整体二分模板,整体二分天然支持修改

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+10;
struct node{
    int op,x,y,k,id;
}q[N],lq[N],rq[N];
int ans[N];
int tr[N];
int a[N];
int lowbit(int x){  
    return x&-x;
}
void add(int x,int c){
    int i;
    for(i=x;i<=N;i+=lowbit(i)){
        tr[i]+=c;
    }
}
int sum(int x){
    int res=0;
    int i;
    for(i=x;i;i-=lowbit(i)){
        res+=tr[i];
    }
    return res;
}
void solve(int vl,int vr,int ql,int qr){
    if(ql>qr){
        return ;
    }
    if(vl==vr){
        for(int i=ql;i<=qr;i++){
            if(q[i].op==2)
            ans[q[i].id]=vl;
        }
        return ;
    }
    int mid=vl+vr>>1;
    int tmp1=0,tmp2=0;
    for(int i=ql;i<=qr;i++){
        if(q[i].op!=2){
            if(q[i].x<=mid) add(q[i].y,q[i].op),lq[++tmp1]=q[i];
            else
                rq[++tmp2]=q[i];
        }
        if(q[i].op==2){
            int n=sum(q[i].y)-sum(q[i].x-1);
            if(n>=q[i].k){
                lq[++tmp1]=q[i];
            }
            else{
                q[i].k-=n;
                rq[++tmp2]=q[i];
            }
        }
    }
    for(int i=ql;i<=qr;i++){
        if(q[i].op!=2){
            if(q[i].x<=mid)
            add(q[i].y,-q[i].op);

        }
    }
    for(int i=1;i<=tmp1;i++) q[ql+i-1] = lq[i];
    for(int i=1;i<=tmp2;i++) q[ql+tmp1+i-1] = rq[i];
    solve(vl, mid, ql, ql+tmp1-1);
    solve(mid+1, vr, ql+tmp1, qr);
}
int main(){
    int n,m;
    cin>>n>>m;
    int i;
    for(i=1;i<=n;i++){
        int x;
        scanf("%d",&x);
        a[i]=x;
        q[i]=node{1,x,i};
    }
    int cnt=0;
    for(i=1;i<=m;i++){
        string s;
        cin>>s;
        if(s=="C"){
            int l;
            int x;
            scanf("%d%d",&l,&x);
            q[++n]=node{-1,a[l],l};
            q[++n]=node{1,x,l};
            a[l]=x;
        }
        else{
            int l,r,k;
            scanf("%d%d%d",&l,&r,&k);
            q[++n]=node{2,l,r,k,++cnt};
        }

    }
    solve(-1e9,1e9,1,n);
    for(i=1;i<=cnt;i++)
        printf("%d
",ans[i]);
    return 0;
}
View Code
原文地址:https://www.cnblogs.com/ctyakwf/p/12804474.html