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,1,2,3,4,5], k = 4, x = -1

输出：[1,1,2,3]

提示：

1 <= k <= arr.length

1 <= arr.length <= 104^44

arr 按升序排列

-104^44 <= arr[i], x <= 104^44

相向双指针，我们需要从arr里删去arr.size()-k个元素：

cpp 复制代码

class Solution {
public:
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        int left = 0;
        int right = arr.size() - 1;
        int deleteNum = 0;
        while (left < right) {
            if (deleteNum == arr.size() - k) {
                break;
            }

            if (abs(arr[left] - x) > abs(arr[right] - x)) {
                ++left;
            } else {
                --right;
            }

            ++deleteNum;
        }

        return vector<int>(arr.begin() + left, arr.begin() + right + 1);
    }
};

如果arr的长度为n，则此算法时间复杂度为O(n - k)，空间复杂度为O(1)。