lintcode[612]:k个最近的点

给定一些 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;
    }
};