找到k个最接近的元素

问题描述

给定一个排序好的数组，两个整数 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 java.lang.Math;
import java.util.List;
import java.util.ArrayList;
import java.util.Collections;

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 < result.size(); i++){
		System.out.println(result.get(i));
		}
	}
	
	public static List<Integer> solution(int[] nums,int k, int x){
		if(null == nums || nums.length == 0){
			return Collections.emptyList();
		}

		int begin = 0;
		int end = nums.length - 1;
		while(end - begin + 1 > k){
			if(Math.abs(x - nums[end]) >= Math.abs(x - nums[begin])){
				end--;
			}else{
				begin++;
			}
		}

		List<Integer> result = new ArrayList<Integer>();
		for(int i = begin; i <= end; i++){
			result.add(nums[i]);
		}

		return result;
	}
}

 

解法二

    public List<Integer> findClosestElements(int[] nums, int k, int x) {
	    List<Integer> res = new ArrayList<>();
	    int start = 0, end = nums.length-k;
	    while(start<end){
	        int mid = start + (end-start)/2;
	        if(Math.abs(nums[mid]-x) > Math.abs(nums[mid + k] - x)){  
	            start = mid+1;
	        } else {
	            end = mid;
	        }
	    }
	    for(int i=start; i<start+k; i++){
	       res.add(nums[i]);
	    }
		
	    return res;
	}