入门力扣自学笔记134 C++ （题目编号658）

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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

思路：

一道简单的优先级队列的问题。

1.建立一个大顶堆,并维护大顶堆的大小为K

2.将最后大顶堆里面的元素转移到数组里面，并对数组进行排序。

---

代码：

class Solution {
public:
    vector<int> findClosestElements(vector<int>& arr, int k, int x) {
        //建立一个大顶堆
        priority_queue<pair<int,int>>temp;
        for(int i=0;i<arr.size();i++){
            //维护大顶堆的的大小为K
            if(temp.size()<k){
                temp.push({abs(arr[i]-x),i});
            }
            else{
                if(abs(arr[i]-x)<abs(temp.top().first)){
                    temp.pop();
                    temp.push({abs(arr[i]-x),i});
                }
            }
        }       
        vector<int>res;
        while(!temp.empty()){
            res.push_back(arr[temp.top().second]);
            temp.pop();
        }
        //对结果进行排序
        sort(res.begin(),res.end());
        return res;
    }
};