leetcode_middle_19_378. Kth Smallest Element in a Sorted Matrix

最新推荐文章于 2025-08-08 21:26:37 发布

pusude

最新推荐文章于 2025-08-08 21:26:37 发布

阅读量172

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/pusude/article/details/54880617

leetcode 专栏收录该内容

183 篇文章

订阅专栏

本文介绍了一种使用优先队列解决二维升序矩阵中查找第K小元素的方法。通过不断更新容量为K的优先队列，最终找到所需的第K小元素。

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

题意：

给一个二维数组，表示一个n*n矩阵，每个行和列都升序，找出第k小的数

分析：
一直找规律，写代码，出错，发现规律出错，再改再写再错...反复多次没法，还是用优先队列吧。只要不停地动态更新k大小的优先队列，就能够选出前k个小的，取出队列头就能得到第k大的。

public class Solution {
    public int kthSmallest(int[][] matrix, int k) {
        PriorityQueue<Integer> pq= new PriorityQueue(k, Collections.reverseOrder());  //从小到大
    
        for(int i=0;i < matrix.length;i++){
            for(int j=0;j < matrix.length;j++){
                if(pq.size() < k)
                    pq.add(matrix[i][j]);
                else{
                    if(pq.peek() > matrix[i][j]){  //有更小的就更新
                        pq.poll();
                        pq.offer(matrix[i][j]);
                    }
                }
            }
        }
        return pq.poll();
    }
}