我们有一个由平面上的点组成的列表 points
。需要从中找出 K
个距离原点 (0, 0)
最近的点。
(这里,平面上两点之间的距离是欧几里德距离。)
你可以按任何顺序返回答案。除了点坐标的顺序之外,答案确保是唯一的。
示例 1:
输入:points = [[1,3],[-2,2]], K = 1
输出:[[-2,2]]
解释:
(1, 3) 和原点之间的距离为 sqrt(10),
(-2, 2) 和原点之间的距离为 sqrt(8),
由于 sqrt(8) < sqrt(10),(-2, 2) 离原点更近。
我们只需要距离原点最近的 K = 1 个点,所以答案就是 [[-2,2]]。
示例 2:
输入:points = [[3,3],[5,-1],[-2,4]], K = 2
输出:[[3,3],[-2,4]]
(答案 [[-2,4],[3,3]] 也会被接受。)
提示:
1 <= K <= points.length <= 10000
-10000 < points[i][0] < 10000
-10000 < points[i][1] < 10000
思路一:使用排序算法,Python可以一行代码实现
class Solution(object):
def kClosest(self, points, K):
"""
:type points: List[List[int]]
:type K: int
:rtype: List[List[int]]
"""
return sorted(points, key= lambda p: p[0]**2+p[1]**2)[:K]
C++ 维护一个最大堆,时间复杂度O(nlogk)
static const int _ = []() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
return 0;
}();
bool myCmp(vector<int> a, vector<int> b){
return a[0]*a[0]+a[1]*a[1] < b[0]*b[0]+b[1]*b[1];
}
int getdistance(vector<int> &a){
return a[0]*a[0]+a[1]*a[1];
}
class Solution {
public:
vector<vector<int>> kClosest(vector<vector<int>>& points, int K) {
vector<vector<int>> res;
priority_queue<vector<int>, vector<vector<int>>,
function<bool(vector<int> ,vector<int> )>> pq(myCmp);
for(int i=0;i<points.size();i++){
if(pq.size()==K){
vector<int> p = pq.top();
if(getdistance(points[i])<getdistance(p)){
pq.pop();
pq.push(points[i]);
}
}
else{
pq.push(points[i]);
}
}
while(!pq.empty()){
res.push_back(pq.top());
pq.pop();
}
return res;
}
};