给定一个数组 points
,其中 points[i] = [xi, yi]
表示 X-Y 平面上的一个点,并且是一个整数 k
,返回离原点 (0,0)
最近的 k
个点。
这里,平面上两点之间的距离是 欧几里德距离( √(x1 - x2)2 + (y1 - y2)2
)。
你可以按 任何顺序 返回答案。除了点坐标的顺序之外,答案 确保 是 唯一 的。
示例 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]] 也会被接受。)
class Solution {
public:
vector<vector<int>> kClosest(vector<vector<int>>& points, int k) {
sort(points.begin(), points.end(), [](const vector<int>& p1, const vector<int>& p2) {
return (p1[0] * p1[0] + p1[1] * p1[1]) < (p2[0] * p2[0] + p2[1] * p2[1]);
}); // 自定义排序,根据距离排序
points.resize(k);
return points;// 返回前 k 个点
}
};
[](const vector<int>& p1, const vector<int>& p2)
是一种lambda 表达式(匿名函数)的语法,通常用于定义自定义比较函数。
在sort
的比较函数中,return
语句用于返回两个元素的排序关系
points.resize(k);
是用来调整 points
容器的大小的方法。具体来说,resize
方法可以改变一个 vector
的大小,以容纳指定数量的元素,如果小于当前大小,vector
将被截断。如果大于当前大小,则会增加新元素。这里相当于vector<vector<int>>(points.begin(), points.begin() + k);