问题描述
给定一个排序好的数组,两个整数 k 和 x ,从数组中找到最靠近 x (两数之差最小)的 k 个数。返回的结果必须要是按升序排好的。如果有两个数与 x 的差值一样,优先选择数值较小的那个数。
示例 1:
输入: [1,2,3,4,5], k=4, x=3
输出: [1,2,3,4]
示例 2:
输入: [1,2,3,4,5], k=4, x=-1
输出: [1,2,3,4]
说明:
k 的值为正数,且总是小于给定排序数组的长度。
数组不为空,且长度不超过 10 4
数组里的每个元素与 x 的绝对值不超过 10 4
更新(2017/9/19):
这个参数 arr 已经被改变为一个 整数数组 (而不是整数列表)。 请重新加载代码定义以获取最新更改。
解法一
import ;
import ;
import ;
import ;
class Solution {
public static void main(String[] args) {
int[] tmp = new int[]{1,2,3,4,5};
int k = 4;
int x = -1;
List<Integer> result = solution(tmp, k, x);
for(int i = 0; i < (); i++){
((i));
}
}
public static List<Integer> solution(int[] nums,int k, int x){
if(null == nums || == 0){
return ();
}
int begin = 0;
int end = - 1;
while(end - begin + 1 > k){
if((x - nums[end]) >= (x - nums[begin])){
end--;
}else{
begin++;
}
}
List<Integer> result = new ArrayList<Integer>();
for(int i = begin; i <= end; i++){
(nums[i]);
}
return result;
}
}
解法二
public List<Integer> findClosestElements(int[] nums, int k, int x) {
List<Integer> res = new ArrayList<>();
int start = 0, end = -k;
while(start<end){
int mid = start + (end-start)/2;
if((nums[mid]-x) > (nums[mid + k] - x)){
start = mid+1;
} else {
end = mid;
}
}
for(int i=start; i<start+k; i++){
(nums[i]);
}
return res;
}