「力扣」第 973 题：最接近原点的 K 个点（优先队列）

最新推荐文章于 2023-01-02 21:50:20 发布

liweiwei1419

最新推荐文章于 2023-01-02 21:50:20 发布

阅读量498

点赞数

分类专栏：力扣文章标签：优先队列

本文链接：https://blog.csdn.net/lw_power/article/details/106442926

版权

力扣专栏收录该内容

142 篇文章 19 订阅

订阅专栏

「力扣」第 973 题：最接近原点的 K 个点（优先队列）

链接：https://leetcode-cn.com/problems/k-closest-points-to-origin/

Java 代码：

import java.util.Arrays;
import java.util.PriorityQueue;

public class Solution {

    public int[][] kClosest(int[][] points, int K) {
        int len = points.length;
        if (len == 0) {
            return new int[0][0];
        }

        int[][] res = new int[K][2];

        PriorityQueue<int[]> maxHeap = new PriorityQueue<>(K + 1, (point1, point2) -> (point2[0] * point2[0] + point2[1] * point2[1])
                - (point1[0] * point1[0] + point1[1] * point1[1]));

        for (int[] point : points) {
            maxHeap.offer(point);
            if (maxHeap.size() > K) {
                maxHeap.poll();
            }
        }

        for (int i = 0; i < K; i++) {
            res[i] = maxHeap.poll();
        }
        return res;
    }
}