LintCode: K个最近的点

最新推荐文章于 2021-01-21 09:18:17 发布

Wison618

最新推荐文章于 2021-01-21 09:18:17 发布

阅读量358

点赞数

分类专栏：算法题

本文链接：https://blog.csdn.net/A__helloworld/article/details/77915106

版权

算法题专栏收录该内容

2 篇文章 0 订阅

订阅专栏

注意：计算平方用Math类里面的pow方法计算，但是这个返回的是一个double类型的数字，如果在lintcode上强制转换为int的话，它会提示精度损失，不给通过，所以我就用两个double相减，如果得出的值大于0，就返回1，如果小于0，就返回-1，如果为0，就返回0；

public static Point[] kClosest(Point[] points, Point origin, int k) {

    Comparator<Point> comparator = new Comparator<Point>(){
    @Override
    public int compare(Point a, Point b){
        double disA = Math.pow(a.x - origin.x, 2) + Math.pow(a.y - origin.y, 2);
        double disB = Math.pow(b.x - origin.x, 2) + Math.pow(b.y - origin.y, 2);
        double com = disB - disA;
        if ( com == 0) {
            com = b.x - a.x;
            if (com == 0) {
                com = b.y - a.y;
            }
        }  
        if (com > 0) {
            return 1;
        }
        else if (com == 0) {
            return 0;
        }
        else {
            return -1;
        }
    }
    };
PriorityQueue<Point> pq = new PriorityQueue<Point>(k, comparator);        
        for (Point p : points) {
            pq.offer(p);
            if (pq.size() > k) {
                pq.poll();
            }
        }
        Point[] sortPoints = new Point[k];
        for (int i = k - 1; i >= 0; i--) {
            sortPoints[i] = pq.poll();
        }
        return sortPoints;
    }