给定一些 points 和一个 origin,从 points 中找到 k 个离 origin 最近的点。按照距离由小到大返回。如果两个点有相同距离,则按照x值来排序;若x值也相同,就再按照y值排序。
样例 给出 points = [[4,6],[4,7],[4,4],[2,5],[1,1]], origin = [0, 0], k = 3 返回 [[1,1],[2,5],[4,4]]
思路:想法并不难,遍历数组找到距离的一个排序即可,主要想使用一下数据结构multiset,以及set的自定义排序规则。
参考代码:
/** * Definition for a point. * struct Point { * int x; * int y; * Point() : x(0), y(0) {} * Point(int a, int b) : x(a), y(b) {} * }; */ class Solution { public: /** * @param points a list of points * @param origin a point * @param k an integer * @return the k closest points */ struct mycomp { bool operator () (const Point &a, const Point &b) const { if(a.x * a.x + a.y * a.y == b.x * b.x + b.y * b.y){ if(a.x == b.x){ return a.y < b.y; } return a.x < b.x; } return a.x * a.x + a.y * a.y < b.x * b.x + b.y * b.y; } }; vector<Point> kClosest(vector<Point>& points, Point& origin, int k) { // Write your code here multiset<Point, mycomp> set_; vector<Point> ans; int len = points.size(); for(int i = 0; i < len; i ++){ points[i].x -= origin.x; points[i].y -= origin.y; set_.insert(points[i]); } multiset<Point, mycomp>::iterator itor = set_.begin(); for(int i = 0; i < k; i ++){ Point p ((*itor).x + origin.x, (*itor).y + origin.y); ans.push_back(p); itor++; } return ans; } };