LeetCode 高级 - Kth Smallest Element in a Sorted Matrix

Kth Smallest Element in a Sorted Matrix

给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第k小的元素。

请注意，它是排序后的第k小元素，而不是第k个元素。

示例:

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

返回 13。

说明:

你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。

分析

方法一：简单粗暴，遍历K次，每次找最小的值，将其赋值为MAX，这样第K次遍历找到的就是第K小的值

方法二：利用小顶堆（Java中的PriorityQueue优先队列），第K个出队的即为第k小的值

代码

    class Solution {

        public int kthSmallest(int[][] matrix, int k) {
            PriorityQueue<Integer> queue = new PriorityQueue<>();
            for(int i=0;i<matrix.length;i++){
                for(int j=0;j<matrix[0].length;j++){
                    queue.offer(matrix[i][j]);
                }
            }
            while(--k>0){
                queue.poll();
            }
            return queue.poll();
        }
    }