从今天开始刷LeetCode了,我会经常将做过的题,学到的知识以及一些思路分享出来。
那么看看题:
//题目描述------------------------------------------------------------
我们有一个由平面上的点组成的列表 points。需要从中找出 K 个距离原点 (0, 0) 最近的点。
(这里,平面上两点之间的距离是欧几里德距离。)
你可以按任何顺序返回答案。除了点坐标的顺序之外,答案确保是唯一的。
示例 1:
输入:points = [[1,3],[-2,2]], K = 1
输出:[[-2,2]]
解释:
(1, 3) 和原点之间的距离为 sqrt(10),
(-2, 2) 和原点之间的距离为 sqrt(8),
由于 sqrt(8) < sqrt(10),(-2, 2) 离原点更近。
我们只需要距离原点最近的 K = 1 个点,所以答案就是 [[-2,2]]。
示例 2:
输入:points = [[3,3],[5,-1],[-2,4]], K = 2
输出:[[3,3],[-2,4]]
(答案 [[-2,4],[3,3]] 也会被接受。)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/k-closest-points-to-origin
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
//题目描述结束----------------------------------------------------------------------
由于自己的方法实在太笨,就直接使用官方的解题思路了。
官方的解题思路是使用Sort函数直接排列出距离原点最近的点:
于是我就自己设计了整个测试程序,算法提交也能成功:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class mycomp2 {
public:
bool operator() (const vector<int>& u, const vector<int>& v) {
return u[0] * u[0] + u[1] * u[1] < v[0] * v[0] + v[1] * v[1];
}
};
class Solution {
public:
vector<vector<int> > kClosest(vector<vector<int> >& points, int K) {
sort(points.begin(), points.end(), mycomp2());
return {points.begin(), points.begin() + K};
}
};
int main()
{
Solution s;
vector<vector<int> > points;
vector<vector<int> > answear;
vector<int> a1;
vector<int> a2;
vector<int> a3;
a1.push_back(3);
a1.push_back(3);
a2.push_back(5);
a2.push_back(-1);
a3.push_back(-2);
a3.push_back(4);
points.push_back(a1);
points.push_back(a2);
points.push_back(a3);
answear = s.kClosest(points,2);
//使用循环方式访问,使用Vector嵌套之后,元素的排列类似于二维矩阵
int m = answear.size();
int n = answear[0].size();
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
cout<< answear[i][j]<<" ";
}
cout<<endl;
}
}
学习到的点:
1.我的编译器(Dev-c++)是不支持直接在Sort函数里面像官方那样直接自定义排序方式的。所以需要新建一个类mycom2来自定义排序方式。并且使用这种方法也可以在LeetCode中通过,所以应该是可以的。
2.使用vector作为函数的参数或者返回值时,一定要加&
3.当使用嵌套方式使用Vector时,元素的排列有点类似于二维矩阵
4.自定义排列方式的格式如mycom2类,返回值为bool operator()(参数)
总结:
使用Sort函数的话整体上这题还是不难,还是挺好理解的。至于其它的大小堆和快速排序解题方法,因为对这些不熟,所以最近会回顾排序方法再回来解题的。