ok
其实这题比较简单,有很多种方式
看题:
其实就是比数组中元素的平方和,然后根据大小排序,输出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,就这样,因为你很重要,所以我从不打扰。
加了个油!