每日一题--658. 找到 K 个最接近的元素

题目描述

给定一个 排序好 的数组 arr ，两个整数 k 和 x ，从数组中找到最靠近 x（两数之差最小）的 k 个数。返回的结果必须要是按升序排好的。

整数 a 比整数 b 更接近 x 需要满足：

|a - x| < |b - x| 或者
|a - x| == |b - x| 且 a < b

示例 1：

输入：arr = [1,2,3,4,5], k = 4, x = 3
输出：[1,2,3,4]
示例 2：

输入：arr = [1,2,3,4,5], k = 4, x = -1
输出：[1,2,3,4]

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/find-k-closest-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

答题思路

本人原始答题思路

正向解题，找出目标数组中开始边界和结束边界。
结果，题目分析不够透彻执行失败。

class Solution {
    public List<Integer> findClosestElements(int[] arr, int k, int x) {
       
        List<Integer> list = new ArrayList<>(k);

        int index = 0;
        for (int i=0; i < arr.length; i++) {
            if (x == arr[i]) {
                index = i;
                break;
            }
        }


        // left
        int count = index;
        do {
            list.add(arr[count]);
            if(list.size() == k) {
                Collections.reverse(list);
                return list;
            }
            
            count --;
        } while (count >= 0);
    

        Collections.reverse(list);

        //right
        while(list.size() < k) {
            list.add(arr[index+1]);
            index++;
        }

    
        return list;
    }
}

网友答题思路

逆向思路，总共有n个数，找出不需要的n-k个数，就是目标结果。
数组中数是连续有序，所以找到最左边界，然后向右读k个数就是目标结果

class Solution {
   public List<Integer> findClosestElements(int[] arr, int k, int x) {
      int length = arr.length;
      int delCnt = length - k;

      // 找出起始下标，从下标开始查找 k个元素
      int left = 0, right = length-1;
      while(delCnt > 0) {
          if (x - arr[left] < arr[right] -x) {
          	// 左边数更接近x，剔除right
              right--;
          } else {
              left ++;
          }
          delCnt--;
      }

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