C练题笔记之：Leetcode-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 <= 104
arr 按 升序 排列
-104 <= arr[i], x <= 104

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

结果：

解题思路：

首先找到数组中最靠近x的值。因为后面是计算开始的值，start是记录left的，所以这里如果没有等于x的，则取比x大一个的那个值。

这样，下一个循环的时候，如果比x小的最大的那个值距离x更近，则会start 左移。始终保持start的位置是我们要记录的最开始的位置。

最后就按照start的位置开始k个数值都放到返回数组里就好了。

代码：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* findClosestElements(int* arr, int arrSize, int k, int x, int* returnSize){
    int *retArr = (int *)malloc(sizeof(int) * k);
    * returnSize = k;
    // 找到最接近x的值，如果没有==x的，就取第一个比x大的值
    int mid = 0;
    for (; mid < arrSize - 1; mid++) {
        if (arr[mid] == x) {
            break;
        } else if (arr[mid] < x && arr[mid + 1] > x) {
            mid++;
            break;
        } else if (arr[mid] > x) {
            break;
        }
    }
    int right, left;
    if (mid == 0) {
        left = mid;
        right = mid + 1;
    } else {
        left = mid - 1;
        right = mid;
    }

    // 通过循环找到需要我们记录的连续数据的最开始那个下标
    int start = mid;
    for (int i = 0; i < k; i ++) {
        if (left < 0) {
            break;
        } else if (right >= arrSize) {
            start = left--;
        } else if(left >= 0 && abs(arr[left] - x) < abs(arr[right] - x)) {
            start = left--;
        } else if (arr[left] < arr[right] && abs(arr[left] - x) == abs(arr[right] - x)) {
            start = left--;
        } else {
            right++;
        }
    }

    for (int i = 0; i < k; i++) {
        retArr[i] = arr[start + i];
    }
    return retArr;
}