https://www.luogu.org/problemnew/show/P2617
题意:给定长度为10万的数组,要求支持两种操作:
1.单点修改
2.查询区间第k小
思路:整体二分https://zhuanlan.zhihu.com/p/55322598

#include<bits/stdc++.h>
using namespace std;
#define lowbit(x) (x&(-x))
#define maxn (100000+100)
#define INF 0x3f3f3f3f

int a[maxn],n,m,cnt,qid,c[maxn],ans[maxn];
struct Query{
    int l,r,k,id,op;
}q[maxn*3],q1[maxn*3],q2[maxn*3];//注意这里开3倍,因为m次操作最差需要2*m,加上n。

int sum(int x)
{
    int ret=0;
    while(x)
    {
        ret+=c[x];
        x-=lowbit(x);
    }
    return ret;
}

void add(int x,int d)
{
    while(x<=n)//维护的是位置下标,故到n
    {
        c[x]+=d;
        x+=lowbit(x);
    }
}

void solve(int l,int r,int L,int R){
    if(L > R) return ;
    if(l == r){
        for(int i=L;i<=R;i++) 
            if(q[i].op==2) ans[q[i].id]=l;
        return ; 
    }
    int mid=(l+r)>>1,cnt1=0,cnt2=0,x;
    for(int i=L;i<=R;i++){
        if(q[i].op==1){
            if(q[i].l <= mid) q1[++cnt1]=q[i],add(q[i].id,q[i].r);
            else q2[++cnt2]=q[i];
        }
        else {
            x=sum(q[i].r)-sum(q[i].l-1);
            if(q[i].k <= x) q1[++cnt1]=q[i];
            else q[i].k-=x,q2[++cnt2]=q[i];
        }
    }
    for(int i=1;i<=cnt1;i++)
        if(q1[i].op==1) add(q1[i].id,-q1[i].r);
    for(int i=1;i<=cnt1;i++) q[L+i-1]=q1[i];
    for(int i=1;i<=cnt2;i++) q[L+i+cnt1-1]=q2[i];
    solve(l,mid,L,L+cnt1-1);
    solve(mid+1,r,L+cnt1,R);
}

int main()
{
    //freopen("input.in","r",stdin);
    int x,y,z;
    char cmd[5];
    cin>>n>>m;
    for(int i=1;i<=n;i++)scanf("%d",&a[i]),q[++cnt]=(Query){a[i],1,0,i,1};
    for(int i=1;i<=m;i++)
    {
        scanf("%s",cmd);
        if(cmd[0]=='Q')scanf("%d%d%d",&x,&y,&z),q[++cnt]=(Query){x,y,z,++qid,2};
        else scanf("%d%d",&x,&y),q[++cnt]=(Query){a[x],-1,0,x,1},q[++cnt]=(Query){a[x]=y,1,0,x,1};
    }
    solve(0,INF,1,cnt);
    for(int i=1;i<=qid;i++)printf("%d\n",ans[i]);
    return 0;
}