hdu6621 K-th Closest Distance(二分+主席树)

题目链接

大意：给你一个长度为n的数组，m个询问，每次询问给你一个区间$[l,r]$，给你一个p
设长度为$r-l+1$的数组S，其中$S_i=|p-a_{l+i-1}|,i\in [1,r-l+1]$，让你输入数组S的第k小。
思路：对给定的数组建一颗主席树；
然后每次二分答案，然后check每次的mid，区间$[p-mid,p+mid]的元素个数是否大于k即可$
细节见代码

#include <bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mp make_pair
#define pb push_back

using namespace std;

LL gcd(LL a,LL b){return b?gcd(b,a%b):a;}
LL lcm(LL a,LL b){return a/gcd(a,b)*b;}
LL powmod(LL a,LL b,LL MOD){LL ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;}
const int N = 2e6 +11;
struct uzi
{
	int sum,l,r;
}T[N*40];
int now,a[N],b[N],t[N];
void update(int l,int r,int &x,int y,int pos){
	T[++now]=T[y];T[now].sum++;x=now;
	if(l==r)return ;
	int mid=l+r>>1;
	if(mid>=pos)update(l,mid,T[x].l,T[y].l,pos);
	else update(mid+1,r,T[x].r,T[y].r,pos);
}
int Ta,n,m;
int c(int l,int r,int x,int y,int L,int R){
	if(l>=L&&R>=r)return T[y].sum-T[x].sum;
	int mid=l+r>>1;
	int ans=0;
	if(L<=mid)ans+=c(l,mid,T[x].l,T[y].l,L,R);
	if(R>mid)ans+=c(mid+1,r,T[x].r,T[y].r,L,R);
	return ans;
}
int main(){
	for(scanf("%d",&Ta);Ta;Ta--){
		scanf("%d%d",&n,&m);
		now=0;
		for(int i=1;i<=n;i++)scanf("%d",&a[i]);
		for(int i=1;i<=n;i++)update(1,2000000,t[i],t[i-1],a[i]);
		int x=0;
		for(int i=1;i<=m;i++){
			int l,r,p,k;
			scanf("%d%d%d%d",&l,&r,&p,&k);
			l^=x;
			r^=x;
			p^=x;
			k^=x;
			int L=0,R=1e6+1;
			int ans=0;
			while(L<=R){
				int mid=L+R>>1;
				int ri=p+mid;
				int le=p-mid;
				le=max(1,le);
				ri=min(ri,2000000);
				if(c(1,2000000,t[l-1],t[r],le,ri)>=k){
					ans=mid;
					R=mid-1;
				}else L=mid+1;
			}
			x=ans;
			printf("%d\n",x );
		}
	}
	return 0;
}