2019 Multi-University Training Contest 4 K-th Closest Distance 二分+主席树

题意：给你n个数，每次查询l，r区间内与p距离第k大的数

题解：二分+主席树（其实用的是权值），每次查询p-mid和p+mid之间的个数是不是k。。。傻逼了数组开小了。。然后查询也查小了，卡了好长时间

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<vector>
#include<set>
#include<map>
#include<stack>
#include<queue>
#include<time.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define ms(s) memset(s, 0, sizeof(s))
const int INF = 0x3f3f3f3f;
const int maxn = 1e6+5;
struct node {
    int l,r;
    int sum;
}T[maxn*20];
int rt[maxn],a[maxn];
vector<int>v;
int n,m,tot;
void init() {
    rt[0] = tot = T[0].l = T[0].r =T[0].sum = 0;
}
void build(int &p,int l,int r){
    p=++tot;
    T[p].sum=0;
    if(l==r)return ;
    int mid=(l+r)/2;
    build(T[p].l,l,mid);
    build(T[p].r,mid+1,r);
}
void update(int l,int r,int &now,int last,int k){
    T[++tot]=T[last];
    now=tot;
    T[tot].sum++;
    if(l==r)return;
    int mid=(l+r)/2;
    if(k<=mid){
        update(l,mid,T[now].l,T[last].l,k);
    }else update(mid+1,r,T[now].r,T[last].r,k);
}

int query(int l,int r,int L,int R,int ql,int qr){
    if(l>=ql&&r<=qr)return T[R].sum-T[L].sum;
    int mid=(l+r)/2;
    int res=0;
    if(ql<=mid)res+=query(l,mid,T[L].l,T[R].l,ql,qr);
    if(qr>mid)res+=query(mid+1,r,T[L].r,T[R].r,ql,qr);
    return res;
}
int main() 
{
    int t;
    scanf("%d",&t);
    while(t--){
        init();
        scanf("%d%d",&n,&m);
        build(rt[0],1,n);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            update(1,1000000,rt[i],rt[i-1],a[i]);
        }
        int l,r,p,k,las=0;
        while(m--){
            scanf("%d%d%d%d",&l,&r,&p,&k);
            l^=las,r^=las,p^=las,k^=las;
            int L=0,R=1000000,ans;
            while(L<=R){
                int mid=(L+R)/2;
                int num=query(1,1000000,rt[l-1],rt[r],p-mid,p+mid);
                if(num>=k){
                    ans=mid;
                    R=mid-1;
                }else L=mid+1;
            }
            printf("%d\n",ans);
            las=ans;
        }
    }
}