leetcode973 最接近原点的 K 个点

我们有一个由平面上的点组成的列表 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]]。

解题思路：
定义一个比较函数，按照距离原点距离从小到大返回
此处在类内定义cmp，需要定义为静态static（静态函数没有this指针），因为成员函数都有一个隐含的this指针，sort函数调用cmp函数的时候只用到两个参数，发生形参与实参不匹配。

class Solution {
public:
    static bool cmp(vector<int>& a, vector<int>& b){
        return (a[0] * a[0] + a[1] * a[1]) < (b[0] * b[0] + b[1] * b[1]);
    }
    vector<vector<int>> kClosest(vector<vector<int>>& points, int K) {
        vector<vector<int>> res;
        sort(points.begin(), points.end(), cmp);
        points.resize(K);
        return points;
    }
};

还有一种办法：
直接调用STL中的nth_element函数。nth_element(a+l,a+k,a+r)会使a这个数组中区间[l,r)内的第k小的元素处在第k个位置上。

class Solution {
public:
    vector<vector<int>> kClosest(vector<vector<int>>& points, int K) {
        nth_element(points.begin(),points.begin()+K,points.end(), 
        [](auto&& l, auto&& r){
            return l[0]*l[0]+l[1]*l[1]<r[0]*r[0]+r[1]*r[1];
            });
        points.resize(K);
        return points;
    }
};

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/k-closest-points-to-origin
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。