LeetCode 973. 最接近原点的 K 个点

最新推荐文章于 2024-04-19 04:11:06 发布

庾信平生最萧瑟

最新推荐文章于 2024-04-19 04:11:06 发布

阅读量137

点赞数

分类专栏： LeetCode 文章标签：排序

本文链接：https://blog.csdn.net/qq_44669578/article/details/107758098

版权

LeetCode 专栏收录该内容

301 篇文章 2 订阅

订阅专栏

原题目： https://leetcode-cn.com/problems/k-closest-points-to-origin/

思路：

采用封装好的nth_element方法

代码：

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