2019 Multi-University Training Contest 4 - K-th Closest Distance

链接:http://acm.hdu.edu.cn/showproblem.php?pid=6621

 

参考的blog:     https://blog.csdn.net/your_eyes_see_sea/article/details/77801856

&&         https://blog.csdn.net/ratina/article/details/98090625

代码如下:

#include <iostream>
#include <stdio.h>
#include <algorithm> 
#define LL long long
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=1e5+50;
const int maxnum=1e6+50;

int n,m,a[maxn];
int root[maxn],t[maxnum<<5],ls[maxnum<<5],rs[maxnum<<5],cnt;
int build(int l,int r) {
    int root = ++cnt;
    if(l==r) {
        t[root]=0;
        return root;
    }
    int mid=(l+r)>>1;
    ls[root]=build(l,mid);
    rs[root]=build(mid+1,r);
    t[root]=t[ls[root]]+t[rs[root]];
    return root;
}

int updata(int l,int r,int v,int pre) {
    int rt=++cnt;
    if(l==r) {
        t[rt]=t[pre]+1;
        return rt;
    }
    ls[rt]=ls[pre],rs[rt]=rs[pre];
    int mid=(l+r)>>1;
    if(v<=mid)
        ls[rt]=updata(l,mid,v,ls[pre]);
    else
        rs[rt]=updata(mid+1,r,v,rs[pre]);
    t[rt]=t[ls[rt]]+t[rs[rt]];
    return rt;
}
int query(int rt1,int rt2,int l,int r,int ql,int qr) {
    if(ql<=l&&r<=qr)
        return t[rt2]-t[rt1];
    if(l>qr||r<ql)
        return 0;
    int mid=(l+r)>>1;
    return query(ls[rt1],ls[rt2],l,mid,ql,qr)+query(rs[rt1],rs[rt2],mid+1,r,ql,qr);
}
int L,R,P,K,X;

int main() {      //主席树(不太懂^~^``) + 二分 
    int T;
    scanf("%d",&T);
    while(T--) {
        scanf("%d %d",&n,&m);
        int l=maxa,r=0;
        for(int i=1; i<=n; i++) {
            scanf("%d",&a[i]);
            l=min(l,a[i]);     //l 最小值  r最大值 
            r=max(r,a[i]);
        }
        cnt=0;
        
        root[0]=build(l,r);   //建旧树
         
        for(int i=1; i<=n; i++){     
            root[i]=updata(l,r,a[i],root[i-1]); //在旧树上建立n课树，对于树tree[i]插入的是[1,i]。
            //                 这样在求区间[l,r]时,tree[r] - tree[l-1]就是我们要找的树，遍历这颗树就行了 
        } 
        
        X=0;
        while(m--) {
            scanf("%d %d %d %d",&L,&R,&P,&K);
            L=L^X;
            R=R^X;
            P=P^X; 
            K=K^X;
            int low=0,up=1e6,mid;      /*[p-x,p+x]是否有k个;  >=k 令up = mid-1 ; <k  low = mid+1   ans is 最后一次满足 >= k
            */
            while(low<=up) {  //二分查找 
                int mid=(low+up)>>1;
                if(query(root[L-1],root[R],l,r,max(l,P-mid),min(r,P+mid))>=K) {
                    X=mid;             //经过updata的新树,   root[L-1]是新树的根结点;
                    up=mid-1;  
                } else
                    low=mid+1;
            }
            printf("%d\n",X);
        }
    }
    return 0;
}