LeetCode—378. Kth Smallest Element in a Sorted Matrix

最新推荐文章于 2022-10-28 23:16:46 发布

原创最新推荐文章于 2022-10-28 23:16:46 发布 · 3.1k 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

226 篇文章

订阅专栏

本文介绍了一种使用最大堆解决寻找有序矩阵中第K小元素的问题的方法，并提供了详细的Java实现代码。

Kth Smallest Element in a Sorted Matrix思路：1.首先这题把所有的数组遍历出来然后直接排序就可以AC..... 2.正常思路就是维系最大堆。

GitHub地址：https://github.com/corpsepiges/leetcode

目前java版本的答案大约进度是免费的差40题，python大约是一半，其他的等以后再补充。

点此进入如果可以的话，请点一下star，谢谢。

public class Solution {
    public int kthSmallest(int[][] matrix, int k) {
        PriorityQueue<Integer> maxHeap = new PriorityQueue<Integer>(new Comparator<Integer>() {
            public int compare(Integer a0, Integer a1) {
                if (a0 > a1) {
                    return -1;
                } else if (a0 < a1) {
                    return 1;
                }
                return 0;
            }
        });// 最大堆
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix.length; j++) {
                if ((i + 1) * (j + 1) > k) {
                    break;
                }
                maxHeap.offer(matrix[i][j]);
                if (maxHeap.size() > k) {
                    maxHeap.poll();
                }
            }
        }
        return maxHeap.peek();
    }
}