找到k个最接近的元素

时间:2025-01-19 11:37:30

问题描述

给定一个排序好的数组,两个整数 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;
	}