Leetcode 378 Kth Smallest Element in a Sorted Matrix

最新推荐文章于 2021-01-04 22:26:48 发布

pessis1

最新推荐文章于 2021-01-04 22:26:48 发布

阅读量210

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/pessis1/article/details/73530289

Leetcode 专栏收录该内容

167 篇文章

订阅专栏

本文介绍了一种高效查找n x n矩阵中第k小元素的方法，该矩阵的每一行和每一列都按升序排列。通过使用优先队列实现算法，确保了找到指定位置的元素，同时附带了完整的Java代码实现。

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

Leetcode 375 这个问题和375是同一个问题

可以用完全一样的方法来做

Given a n x n matrix where each of the rows and columns are sorted in ascending order, find the kth smallest element in the matrix.

Note that it is the kth smallest element in the sorted order, not the kth distinct element.

Example:

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

return 13.

Note:
You may assume k is always valid, 1 ≤ k ≤ n2.

public class Solution {
    public int kthSmallest(int[][] matrix, int k) {
        PriorityQueue<Pairs> q = new PriorityQueue<>();
        
            for(int j = 0; j < matrix[0].length; j++){
                q.offer(new Pairs(matrix[0][j], 0, j));
            }
            
        int count = 0;
        while(!q.isEmpty() &&  count++ < k - 1){
           Pairs tmp = q.poll();
           if(tmp.x == matrix.length - 1) continue;
           q.offer(new Pairs(matrix[tmp.x + 1][tmp.y], tmp.x + 1, tmp.y));
        }
        return q.poll().val;
    }
    
    class Pairs implements Comparable<Pairs> {//这里就是用了一个class
    int x, y, val;
    public Pairs (int val, int x, int y) {
        this.x = x;
        this.y = y;
        this.val = val;
    }
    
    @Override
    public int compareTo (Pairs that) {
        return this.val - that.val;
    }
}
}