2019杭电暑假多校4：K-th Closest Distance【主席树+二分】

题目：

2019杭电暑假多校4：K-th Closest Distance

题意:

给定一个长度为N的数组，M次在线询问区间【L,R】中 abs(a[i] - p) 第 k 小的值是多少

分析：

如果将区间排好序后，abs(a[i] - p) 的值一定凹的，从 p 开始越往两边走就越大；考虑二分答案，如区间【L，R】的数中位于【p-mid，p+mid】的个数小于等于 k，那么这个答案一定小于等于mid，否则一定大于mid；查找区间内位于【p-mid,p+mid】数的个数用主席树查找即可

代码:

#include <bits/stdc++.h>

using namespace std;
const int maxm = 1e5+15;
const int maxn = 1e6+15;
int cnt,n,m,v,L,R,k,p,root[maxm];
struct tree{
    int l,r,num;
}T[maxn*22];
void build(int l,int r,int &x,int y,int v){
    T[++cnt] = T[y]; T[cnt].num++; x = cnt;
    if(l == r) return ;
    int mid = (l+r) >> 1;
    if(v > mid) build(mid+1,r,T[x].r,T[y].r,v);
    else build(l,mid,T[x].l,T[y].l,v);
}
void query(int l,int r,int ql,int qr,int L,int R,int &sum){
    if(l > qr || r < ql) return ;
    if(l >= ql && r <= qr){
        sum += T[R].num-T[L].num;
        return ;
    }
    int mid = (l+r) >> 1;
    query(l,mid,ql,qr,T[L].l,T[R].l,sum);
    query(mid+1,r,ql,qr,T[L].r,T[R].r,sum);
}
int main(){
    int Case; scanf("%d",&Case);
    while(Case--){
        cnt = 0;scanf("%d %d",&n,&m);
        for(int i = 1;i <= n; ++i){
            scanf("%d",&v);
            build(1,maxn-1,root[i],root[i-1],v);
        }
        int ans = 0; 
        while(m--){
            scanf("%d %d %d %d",&L,&R,&p,&k);
            L = L^ans; R = R^ans;
            p = p^ans; k = k^ans;
            int ql = 0,qr = maxn-1;
            while(ql <= qr){
                int mid = (ql+qr)>>1,sum = 0;
                query(1,maxn-1,max(1,p-mid),min(p+mid,maxn-1),root[L-1],root[R],sum);
                if(k <= sum) qr = mid-1;
                else ql = mid+1;
            }
            printf("%d\n",ans = ql);
        }
    }
    return 0;
}