力扣：658. 找到 K 个最接近的元素

原创已于 2022-08-25 22:23:12 修改 · 191 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #java

于 2022-08-25 22:23:10 首次发布

算法同时被 2 个专栏收录

278 篇文章

订阅专栏

菜鸟刷题记录

204 篇文章

订阅专栏

* 排序
* 按照离x距离的远近进行排序
* 排序之后，就会按照距离近的在前面，远得在后面，然后直接截取就行
* 截取之后在排序，就是我们的结果啦
* 排序的时间复杂度是logn,遍历添数组是n总共是nlogn
* 空间复杂度是logn

public List<Integer> findClosestElements(int[] arr, int k, int x) {
        List<Integer> res = new ArrayList<>();
        for (int i : arr) {
            res.add(i);
        }

        Collections.sort(res, (a, b) ->{
            if (Math.abs(a - x) != Math.abs(b - x) ){
                return Math.abs(a - x) - Math.abs(b - x);
            }
            return a - b;
        });
        List<Integer> list = res.subList(0, k);
        Collections.sort(list);
        return list;
 }

* 二分解法，双指针
*      1.首先找到最接近x的两个点，分别定义为left和right
*      2.注意，当left小于0 那个只需要加right就行，当right大于数组的长度的时候，只需要减left
*      3.判断那边距离更近，添加哪边

public List<Integer> findClosestElements2(int[] arr, int k, int x) {
        int right = binary(arr,x);
        int left = right - 1;
        while (k-- > 0){
            if (left < 0){
                right++;
            } else if (right >= arr.length ){
                left--;
            } else if (x - arr[left] <= arr[right] - x){
                left--;
            } else {
                right++;
            }
        }

        List<Integer> list = new ArrayList<>();
        for (int i = left + 1; i < right; i++){
            list.add(arr[i]);
        }
        return list;
    }

    private int binary(int[] arr, int x){
        int left = 0, right = arr.length - 1;
        while (left < right){
            int mid = (left + right) >> 1;
            // 如果中位数大于x
            if (arr[mid] >= x){
                right = mid;
            }else { // 如果中位数小于x
                left = mid + 1;
            }
        }
        return left;
    }