LeetCode Daily challenge - K Closest Points to Origin

EricJeffrey

于 2020-06-01 14:28:29 发布

阅读量133

点赞数

分类专栏：算法文章标签：算法 leetcode

本文链接：https://blog.csdn.net/q15037911903/article/details/106471779

版权

算法专栏收录该内容

40 篇文章 0 订阅

订阅专栏

文章目录

题目大意
思路
代码
总结

题目大意

找前K小元素，只不过元素是二维的坐标

思路

很容易想到排序后取前K个，虽然可以但是速度不够快。实际上类似于找第K大(小)，可以用划分的方法找，每次随机划分，然后判断接下来从哪儿一边找。

STL中有partial_sort和nth_element，尽情AC吧。。。。。

代码

第二种方法的代码。

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