Problem Description
You have an array: a1, a2, , an and you must answer for some queries.
For each query, you are given an interval [L, R] and two numbers p and K. Your goal is to find the Kth closest distance between p and aL, aL+1, …, aR.
The distance between p and ai is equal to |p - ai|.
For example:
A = {31, 2, 5, 45, 4 } and L = 2, R = 5, p = 3, K = 2.
|p - a2| = 1, |p - a3| = 2, |p - a4| = 42, |p - a5| = 1.
Sorted distance is {1, 1, 2, 42}. Thus, the 2nd closest distance is 1.
Input
The first line of the input contains an integer T (1 <= T <= 3) denoting the number of test cases.
For each test case:
冘The first line contains two integers n and m (1 <= n, m <= 10^5) denoting the size of array and number of queries.
The second line contains n space-separated integers a1, a2, …, an (1 <= ai <= 10^6). Each value of array is unique.
Each of the next m lines contains four integers L’, R’, p’ and K’.
From these 4 numbers, you must get a real query L, R, p, K like this:
L = L’ xor X, R = R’ xor X, p = p’ xor X, K = K’ xor X, where X is just previous answer and at the beginning, X = 0.
(1 <= L < R <= n, 1 <= p <= 10^6, 1 <= K <= 169, R - L + 1 >= K).
Output
For each query print a single line containing the Kth closest distance between p and aL, aL+1, …, aR.
Sample Input
1
5 2
31 2 5 45 4
1 5 5 1
2 5 3 2
Sample Output
0
1
分析:
其实这个题,我真的思路全会了,可是还是wa了一个晚上。
这个题目就是主席树+二分。
题目范围都可以不离散化。但是我还是离散了一下。
主席树就很正常的按模板建。二分那个差值mid,那么当前需要查询的区间就是p-mid,p+mid。同时,在真正查询的时候,我们要把数值通过lower_bound转换成b里面的下标进行查询。
#include"stdio.h"
#include"string.h"
#include"algorithm"
using namespace std;
int N,M;
int T[200010],L[50 * 200010],R[50 * 200010],sum[50 * 200010],top;
int a[200010],b[200010],len;
void Build_Tree(int &root,int l,int r)
{
root = ++ top;
sum[root] = 0;
int mid = (l + r) >> 1;
if(l < r)
{
Build_Tree(L[root],l,mid);
Build_Tree(R[root],mid + 1,r);
}
}
void Update(int &root,int root1,int l,int r,int x)
{
root = ++ top;
L[root] = L[root1];
R[root] = R[root1];
sum[root] = sum[root1] + 1;
int mid = (l + r) >> 1;
if(l >= r)
return ;
if(x > mid)
Update(R[root],R[root1],mid + 1,r,x);
else
Update(L[root],L[root1],l,mid,x);
}
int query(int x,int y,int ql,int qr,int l,int r)
{
if(ql <= l && qr >= r) return sum[y] - sum[x];
int mid = (l + r) >> 1;
int res = 0;
if(ql <= mid)
res += query(L[x],L[y],ql,qr,l,mid);
if(qr > mid)
res += query(R[x],R[y],ql,qr,mid + 1,r);
return res;
}
int check(int ql,int qr,int l,int r,int k)
{
int t1 = lower_bound(b + 1,b + 1 + len,ql) - b;
int t2 = lower_bound(b + 1,b + 1 + len,qr + 1) - b - 1;
if(t1 > len || t2 > len || t1 > t2) return 0;
int tmp = query(T[l - 1],T[r],t1,t2,1,len);
if(tmp >= k) return 1;
return 0;
}
int main()
{
int t;
scanf("%d",&t);
while(t --)
{
top = 0;
scanf("%d%d",&N,&M);
for(int i = 1; i <= N; i ++)
{
scanf("%d",&a[i]);
b[i] = a[i];
}
sort(b + 1,b + 1 + N);
len = unique(b + 1,b + 1 + N) - b - 1;
Build_Tree(T[0],1,len);
for(int i = 1; i <= N; i ++)
{
int t = lower_bound(b + 1,b + 1 + len,a[i]) - b;
Update(T[i],T[i - 1],1,len,t);
}
int x = 0;
while(M --)
{
int l,r,p,k;
scanf("%d%d%d%d",&l,&r,&p,&k);
l ^= x; r ^= x; p ^= x; k ^= x;
int l1 = 0,r1 = 1e6;
while(l1 + 1 < r1)
{
int mid = (l1 + r1) >> 1;
if(check(p - mid,p + mid,l,r,k) == 1)
r1 = mid;
else
l1 = mid;
}
if(check(p - l1,p + l1,l,r,k) == 1)
x = l1;
else
x = r1;
printf("%d\n",x);
}
}
}