0
点赞
收藏
分享

微信扫一扫

力扣-973题 最接近原点的K个点(C++)- 优先队列

回望这一段人生 2022-04-14 阅读 25
leetcodec++

题目链接:https://leetcode-cn.com/problems/k-closest-points-to-origin/
题目如下:
在这里插入图片描述
在这里插入图片描述

class Solution {
public:
    vector<vector<int>> kClosest(vector<vector<int>>& points, int k) {
        vector<vector<int>> res;
        //默认是less算子也就是返回a<b,也就是小的优先级也小,而greater算子返回a>b,小的优先级高
        priority_queue<pair<int,int>,vector<pair<int,int>>,greater<>> minheap;

        for(int i=0;i<points.size();i++){
            int sum=points[i][0]*points[i][0]+points[i][1]*points[i][1];
            minheap.push({sum,i});
        }

        while(k--){
            res.push_back(points[minheap.top().second]);
            minheap.pop();
        }
        
        return res;
    }
};
举报

相关推荐

0 条评论