[Luogu2617]Dynamic Rankings（整体二分）

Posted 2020-10-23 租酥雨

Luogu

动态区间第K大的整体二分解法

之前学主席树的时候就做了这道题（明明是树套树不是主席树啊），码量挺大而且调了我一个晚上。换成整体二分我半个小时就写完了而且一A。
写起来就是爽。
其实原理很简单，先把修改和询问放在一起，注意这里不能排序，要严格按照时间戳进行处理。
区间的初值也视作一次修改，之后的每次修改要当成两次，即删除原先那个数，再加入新的那个数。
对于当前的一些询问，二分一个答案\(mid\)。处理所有修改的值小于等于\(mid\)的修改操作，用树状数组维护区间内有多少个数的值小于等于\(mid\)，再把这个数与询问的\(k\)比较，选择向左递归（答案更小）还是向右递归（答案更大）。注意向左向右的时候要保证原修改查询的有序性，即对于某一个修改和某一个查询，若他们被分在了相同的一边（左或右），那么他们的相对顺序一定不能改变。

code

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N = 1e5+5;
int gi()
{
    int x=0,w=1;char ch=getchar();
    while ((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
    if (ch=='-') w=0,ch=getchar();
    while (ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return w?x:-x;
}
struct node{
    int opt,x,y,id,k;
}q[N],q1[N],q2[N];
int n,m,cnt,qcnt,a[N],tmp[N],c[N],ans[N];
void modify(int k,int v){while (k<=n) c[k]+=v,k+=k&-k;}
int query(int k){int s=0;while (k) s+=c[k],k-=k&-k;return s;}
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].opt==0) ans[q[i].id]=l;
        return;
    }
    int mid=l+r>>1;
    for (int i=L;i<=R;i++)
        if (q[i].opt==0) tmp[q[i].id]=query(q[i].y)-query(q[i].x-1);
        else if (q[i].y<=mid) modify(q[i].x,q[i].opt);
    for (int i=L;i<=R;i++)
        if (q[i].opt!=0&&q[i].y<=mid) modify(q[i].x,-q[i].opt);
    int t1=0,t2=0;
    for (int i=L;i<=R;i++)
        if (q[i].opt==0)
            if (q[i].k<=tmp[q[i].id]) q1[++t1]=q[i];
            else q[i].k-=tmp[q[i].id],q2[++t2]=q[i];
        else
            if (q[i].y<=mid) q1[++t1]=q[i];
            else q2[++t2]=q[i];
    for (int i=L,j=1;j<=t1;i++,j++) q[i]=q1[j];
    for (int i=L+t1,j=1;j<=t2;i++,j++) q[i]=q2[j];
    solve(L,L+t1-1,l,mid);solve(L+t1,R,mid+1,r);
}
int main()
{
    n=gi();m=gi();
    for (int i=1;i<=n;i++)
    {
        a[i]=gi();
        q[++cnt]=(node){1,i,a[i]};
    }
    for (int i=1;i<=m;i++)
    {
        char ch=getchar();
        while (ch!='C'&&ch!='Q') ch=getchar();
        if (ch=='C')
        {
            int x=gi(),y=gi();
            q[++cnt]=(node){-1,x,a[x]};
            a[x]=y;
            q[++cnt]=(node){1,x,a[x]};
        }
        else
        {
            int l=gi(),r=gi(),k=gi();
            q[++cnt]=(node){0,l,r,++qcnt,k};
        }
    }
    solve(1,cnt,0,1e9);
    for (int i=1;i<=qcnt;i++) printf("%d\n",ans[i]);
    return 0;
}