leetcode 658 | 找到 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 的值为正数,且总是小于给定排序数组的长度。
数组不为空,且长度不超过 104
数组里的每个元素与 x 的绝对值不超过 104

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-k-closest-elements

 思路:

找到最靠近 x 的k个数,感觉不太难,但是实现起来还真的不太简单,于是我们可以转变一下思路,去除最不靠近x的len-k个数。似乎这个比刚刚那个方案好实现,而且得到的是一个连续的子序列,不需要进行排序操作。

使用两个指针,一个表示数组开始,可以表示数组结束,判断距离然后移动两个指针。最终把两个指针区域的数字返回就行了

public List<Integer> findClosestElements(int[] arr, int k, int x) {
    List<Integer> res = new ArrayList<>();
    if(k<=0){ // 特殊值处理
        return res;
    }
    // len 表示arr的长度
    // l 表示数组低位
    // h 表示数组高位
    int len = arr.length;
    int l=0,h=len-1;
    while(len>k){
        // 判断数组地位距离x近还是高位距离x近
        if(Math.abs(arr[l]-x) > Math.abs(arr[h]-x)){
            l++;
        }else {
            h--;
        }
        len--;
    }
    // 制作结果
    for(int i=l;i<=h;i++) {
        res.add(arr[i]);
    }
    return res;
}
发布了80 篇原创文章 · 获赞 55 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/cong____cong/article/details/104905109