LeetCode Daily challenge - K Closest Points to Origin

题目大意

找前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);
    }
};

总结

除非练习，否则能用库就别手写