find k closest point to origin point in 2d plane

最新推荐文章于 2023-08-30 16:17:39 发布

原创最新推荐文章于 2023-08-30 16:17:39 发布 · 517 阅读

0 ·

CC 4.0 BY-SA版权

strStr 专栏收录该内容

360 篇文章

订阅专栏

本文介绍了一种使用优先队列实现的算法，该算法能够有效地找到二维平面上距离原点最近的K个点。通过定义一个比较器来确定点之间的远近关系，并利用优先队列的数据结构特性进行高效筛选。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

public List<Point> findKClosest(Point[] p, int k) {
        List<Point> res = new LinkedList<>();
        PriorityQueue<Point> queue = new PriorityQueue<>(11, new Comparator<Point>(){
            @Override
            public int compare(Point a, Point b){
                return b.x * b.x + b.y * b.y - a.x * a.x - a.y * a.y;
            }
        });
        for (Point point: p) {
            if (queue.size() < k) {
                queue.offer(point);
            } else {
                Point a = queue.poll();
                if (point.x * point.x + point.y * point.y - a.x * a.x - a.y * a.y < 0) {
                    queue.poll();
                    queue.offer(point);
                }
            }
        }
        while (!queue.isEmpty()) {
            res.add(queue.poll());
        }
        return res;
    }
    
    class Point {
        int x;
        int y;
        public Point (int x, int y) {
            this.x = x;
            this.y = y;
        }
    }