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]

提示：

1 <= k <= arr.length
1 <= arr.length <= 10^4
arr 按 升序 排列
-10^4 <= arr[i], x <= 10^4

解题思路

方法一：排序
首先将数组 arr 按照「更接近」的定义进行排序，如果 a 比 b 更接近 x，那么 a 将排在b 前面。排序完成之后，k个最接近的元素就是数组 arr 的前 k个元素，将这 k 个元素从小到大进行排序后，直接返回。
C语言

int g_val;

static inline int cmp1(const void *pa, const void *pb) {
    int a = *(int *)pa;
    int b = *(int *)pb;
    if (abs(a - g_val) != abs(b - g_val)) {
        return abs(a - g_val) - abs(b - g_val);
    } else {
        return a - b;
    }
}

static inline int cmp2(const void *pa, const void *pb) {
    return *(int *)pa - *(int *)pb;
}

int* findClosestElements(int* arr, int arrSize, int k, int x, int* returnSize) {
    g_val = x;
    qsort(arr, arrSize, sizeof(int), cmp1);
    qsort(arr, k, sizeof(int), cmp2);
    int *res = (int *)malloc(sizeof(int) * k);
    memcpy(res, arr, sizeof(int) * k);
    *returnSize = k;
    return res;
}

方法二：二分查找 + 双指针
假设数组长度为 n，注意到数组 arr 已经按照升序排序，我们可以将数组arr 分成两部分，前一部分所有元素 [0,left] 都小于 x，后一部分所有元素 [right,n−1] 都大于等于 x，left 与 right 都可以通过二分查找获得。

left 和 right 指向的元素都是各自部分最接近 x 的元素，因此我们可以通过比较 left 和 right 指向的元素获取整体最接近 x的元素。如果 x −arr[left]≤arr[right]−x，那么将 left 减一，否则将 right 加一。相应地，如果 left 或 right 已经越界，那么不考虑对应部分的元素。

最后，区间 [left+1,right−1] 的元素就是我们所要获得的结果，返回答案既可。

int binarySearch(const int* arr, int arrSize, int x) {
    int low = 0, high = arrSize - 1;
    while (low < high) {
        int mid = low + (high - low) / 2;
        if (arr[mid] >= x) {
            high = mid;
        } else {
            low = mid + 1;
        }
    }
    return low;
}

int* findClosestElements(int* arr, int arrSize, int k, int x, int* returnSize) {
    int right = binarySearch(arr, arrSize, x);
    int left = right - 1;
    while (k--) {
        if (left < 0) {
            right++;
        } else if (right >= arrSize) {
            left--;
        } else if (x - arr[left] <= arr[right] - x) {
            left--;
        } else {
            right++;
        }
    }
    int *res = (int *)malloc(sizeof(int) * (right - left - 1));
    memcpy(res, arr + left + 1, sizeof(int) * (right - left - 1));
    *returnSize = right - left - 1;
    return res;
}

思考总结

1.中等题目对现在我来说有点困难，里面的快排我都还不熟悉
2.方法三：1.若X在数组中间（包括两个端点），直接使用双指针的方法，但是从两端开始排除， 如果x −arr[left]≤arr[right]−x，将right -1，否则left + 1，直至最后剩余right - left = k；将这K个元素放进返回数组中。2.若x在数组左侧，直接返回数组前k个元素；3.若在数组右侧，返回倒数k个元素。