SP3985&poj2104题解

题目大意

给定一个长度为n的序列，m次询问查询给定区间k小值。n≤1e5, m≤1e4, |ai|≤1e9

题解

先读题，发现不仅要查询第k小数，还要在一定的区间内。

首先先去想如果固定查一个序列怎么做，当然可以直接sort，还有一种想法是二分答案。设当前二分的值为mid，统计区间内有多少数≤mid，记为t。如果t≤k就在左区间二分，反之则让t-=k，再在右区间二分，二分后的答案即为最终答案。

再去想如何和区间扯上关系。容易想到其实t可以用树状数组来维护。只需维护序列每个前缀中≤mid的数的个数（就是当a[i]≤mid时add(i,1)），求区间时前缀和相减即可。

回到这题，若朴素地执行m次二分答案，我们就会浪费掉之前询问到的有价值的信息，此时便可以对值域用整体分治的思想求解。我们可以尝试在值域[-1e9,1e9]上进行二分答案。

设当前二分的值为mid，边界为L和R。我们可以把问题看作一个修改n次，查询m次的操作序列。对于每个修改用树状数组的add操作，对于每个询问li,ri,ki,我们用树状数组维护去查找[li , ri]中≤mid的部分。我们然后把原序列中的数进行分类，分别为≤mid和>mid的数，存储入数组al,ar，并将对应部分的询问也分成ql，qr，然后套用之前讲的二分答案的方法去做。具体一点就是每次递归求解(L, mid, al, ql）与（mid+1, R, ar, qr)，递归边界为L==R或q为空。当L==R时便可直接将二分出的答案l作为每个询问的答案。

在程序实现过程中我们还可以直接把al，ar，ql，qr拷贝回a和q，优化空间。时间复杂度O( (N+M)log 2e9 logN )，加上离散化会更快一点。

Code

#include<bits/stdc++.h>
using namespace std;
#define inf 1e9
int n,m,cnt;
struct node{
    int op,x,y,p;
}q[200005],ql[200005],qr[200005];
int ans[100005],c[100005];
int ask(int x){
    int ans=0;
    for(;x;x-=x&-x)ans+=c[x];
    return ans;
}
void add(int x,int y){
    for(;x<=n;x+=x&-x)c[x]+=y;
}
inline int read(){
    int x=0;bool flag=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')flag=0;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    if(flag)return x;
    return ~(x-1);
}
void solve(int l,int r,int st,int ed){//二分递归求解 
    if(st>ed)return;//q为空 
    int mid=(l+r)>>1;
    if(l==r){
        for(int i=st;i<=ed;i++){
            if(q[i].op)ans[q[i].op]=l;
        }
        return;
    }
    int t1=0,t2=0;
    for(int i=st;i<=ed;i++){
        if(q[i].op==0){
            if(q[i].y<=mid)add(q[i].x,1),ql[++t1]=q[i];
            else qr[++t2]=q[i];
        }
        else{
            int t=ask(q[i].y)-ask(q[i].x-1);
            if(t>=q[i].p)ql[++t1]=q[i];
            else q[i].p-=t,qr[++t2]=q[i];
        }
    }
    for(int i=st;i<=ed;i++){//清空树状数组 
        if(q[i].op==0&&q[i].y<=mid){
            add(q[i].x,-1);
        }
    }
    for(int i=1;i<=t1;i++)q[i+st-1]=ql[i];
    for(int i=1;i<=t2;i++)q[i+st+t1-1]=qr[i];
    solve(l,mid,st,st+t1-1);
    solve(mid+1,r,st+t1,ed);
}
int main(){
    n=read();m=read();
    for(int i=1;i<=n;i++){
        int a=read();
        q[++cnt].x=i;q[cnt].y=a;
    }
    for(int i=1;i<=m;i++){
        q[++cnt].x=read();q[cnt].y=read();q[cnt].p=read();
        q[cnt].op=i;//标记操作序号 
    }
    solve(-inf,inf,1,cnt);
    for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
}