在数轴上给出n个点,a1,a2,……an是这n个点在数轴上的位置,现在要你找到一个位置x,使得f(k,x)是最小的
求f(k,x)的步骤是
1,找到一个序列d,d1,d2……dn,di=|ai-x|,也就是说找到的这个序列是表示每个点到x的距离
2,对d序列进行从小到大排序
3,d[k+1]就是答案
如果有多种答案,随意输出一种
输入
首先输入一个t,表示有t次询问,t(1<=t<=2e5)
接下来输入n,k (1 < = n < = 2e5 , 0 < = k < n )
接下来一行输入n个数 a1,a2……an (1<=ai<=1e9)
保证多次询问的n的和不大于2e5
输出
输出t行,每行一个答案是使得f(k,x)最小的x
样例
Input
3
3 2
1 2 5
2 1
1 1000000000
1 0
4
Output
3
500000000
4
题解:题意就是要寻找一个位置x,求得ai到其距离第k+1小的值的最小值。首先对于一个大小为m排好序的数组,若使di的最大值最小,则x应在(am+a1)/2的位置。那么对于本题对ai先按升序排列,在a[i]~a[i+k]的中点为x,则a[i+k]到x的距离就是第k+1小的值,因为[i,i+k]区间外的di肯定dk+1大,那么遍历数组即可,详情见代码。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
int a[N];
int main(){
int t; cin>>t;
while(t--){
int n,k;
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++){
scanf("%d",a+i);
}
ll mx=INT_MAX,ans;
for(int i=1;i+k<=n;i++){
if(mx>a[i+k]-a[i]){
mx=a[i+k]-a[i];
ans=(a[i+k]+a[i])/2;
}
}
printf("%lld\n",ans);
}
return 0;
}