我的力扣算法973-最接近远点的k个点

ok
其实这题比较简单,有很多种方式
看题:
我的力扣算法973-最接近远点的k个点_第1张图片
其实就是比数组中元素的平方和,然后根据大小排序,输出k个小的,就可以了。
首先我们先看这个方式

class Solution {
public:
    vector<vector<int>> kClosest(vector<vector<int>>& points, int K) {
        sort(points.begin(), points.end(), [](const vector<int>& a, const vector<int>& b) //定义了一个伪函数,用来改变排序方式
        {
            return a[0] * a[0] + a[1] * a[1] < b[0] * b[0] + b[1] * b[1];//改变排序方式,是通过平方和的大小进行排序
        });
        return {points.begin(), points.begin() + K};//输出此时排完序后,从第一个到希望到的k个
    }
};

来看这一个,虽然是利用优先队列,其实利用其他容器存放其实也可以,方法都是差不多的

class Solution {
public:
    vector<vector<int>> kClosest(vector<vector<int>>& points, int K) {
        priority_queue<pair<int,int>> q;//这里创建优先队列,优先队列会自动排序,存放的值是平方和与对应数据的组号。
        for(int i=0;i<K;i++){//将点数组中的前k个数组放入队列中,方便之后输出k个点
            q.emplace(points[i][0]*points[i][0]+points[i][1]*points[i][1],i);
        }
        int n=points.size();
        for(int i=K;i<n;i++){//遍历点数组k后面的所有数据
            int dist=points[i][0]*points[i][0]+points[i][1]*points[i][1];//获得点的平方和
            if(dist<q.top().first){//如果它小于队头,就把对头释放掉,把它放入队列。
                q.pop();
                q.emplace(dist,i);
            }
        }
        vector<vector<int>> ans;
        while(!q.empty()){//把队列中k个值通过点数组输出
            ans.push_back(points[q.top().second]);//因为队列中second存放的是组号,通过组号找到点数组中的点放入到ans数组中。
            q.pop();
        }
        return ans;
    }
};

ok,就这样,因为你很重要,所以我从不打扰。
加了个油!

你可能感兴趣的:(力扣算法练习,队列,算法,数据结构,leetcode,c++)