LeetCode 每日一题 2020-11-9 （最接近原点的 K 个点 最小堆和最大堆）

973. 最接近原点的 K 个点

难度中等

我们有一个由平面上的点组成的列表 points。需要从中找出 K 个距离原点 (0, 0) 最近的点。

（这里，平面上两点之间的距离是欧几里德距离。）

你可以按任何顺序返回答案。除了点坐标的顺序之外，答案确保是唯一的。

思路1：

采用最小堆保存各点与原点的距离，最终按从小到大保存距离（也可以采用向量保存之后再排序）

建立map结构，根据各点的距离分配点的位置，最后根据距离的排序结果将各点加入到输出结果中。

    vector<vector<int>> kClosest(vector<vector<int>>& points, int K) {

        priority_queue<int, vector<int>, greater<int> > q;  // 小顶堆

        unordered_map<int,vector<vector<int>>> M;

        vector<vector<int>> ans;

        int temp=0;

        for(auto index:points){

            temp=index[0]*index[0]+index[1]*index[1];

            M[temp].push_back(index);

            q.push(temp);

        }

        while(q.size()){

            for(auto temp:M[q.top()]){

                if(K==ans.size()) return ans;

                ans.push_back(temp);

            }

            q.pop();            

        }

        return ans;

    }

思路2：

根据STL中的：priority_queue<Type, Container, Functional>

Type为数据类型， Container为保存数据的容器，Functional为元素比较方式。

如果不写后两个参数，那么容器默认用的是vector，比较方式默认用operator<，也就是优先队列是大顶堆，队头元素最大

最小堆： priority_queue<Type, vector<Type>, greater<Type> > pq;

最大堆：priority_queue<Type> pq;

将距离和点在列表中的位置组成 pair<int,int>格式的数据结构，按距离排列各pair量，最后根据排序结果，将列表中对应位置的点加入到输出结果中

    vector<vector<int>> kClosest(vector<vector<int>>& points, int K) {

        priority_queue<pair<int,int>, vector<pair<int,int>>, greater<pair<int,int>> > q;  // 小顶堆

        vector<vector<int>> ans;

        for(int i=0;i<points.size();++i){

            q.push({points[i][0]*points[i][0]+points[i][1]*points[i][1], i});

        }

        while(q.size()){

            if(K==ans.size()) return ans;

            ans.push_back(points[q.top().second] );

            q.pop();            

        }

        return ans;

    }