LeetCode#378 有序矩阵中第k小的元素

卑微求个Offer

于 2020-11-10 09:08:34 发布

阅读量145

点赞数

分类专栏：算法类文章标签：堆

本文链接：https://blog.youkuaiyun.com/weixin_40340397/article/details/109591734

版权

算法类专栏收录该内容

70 篇文章

订阅专栏

本文介绍了一种使用小根堆寻找nxn升序矩阵中第k小元素的方法。通过实例演示了如何利用优先队列实现算法，并确保k的有效性在1到n²之间。

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

给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。
请注意，它是排序后的第 k 小元素，而不是第 k 个不同的元素。

示例：

matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,

返回 13。

方法一：小根堆

class Solution {
    public int kthSmallest(int[][] matrix, int k) {
        PriorityQueue<int[]> queue = new PriorityQueue<>(new Comparator<int[]>(){
            public int compare(int[] o1,int[] o2){
                return o1[0] - o2[0];
            }
        });
        int n = matrix.length;
        for(int i=0;i<n;i++){
            queue.offer(new int[]{matrix[i][0],i,0});
        }
        for(int i=0;i<k-1;i++){
            int temp[] = queue.poll();
            if(temp[2] < n - 1){
                queue.offer(new int[]{matrix[temp[1]][temp[2]+1],temp[1],temp[2] + 1});
            }
        }
        return queue.poll()[0];
    }
}