Java中等题-有序矩阵中第k小的元素(力扣)

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

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

你必须找到一个内存复杂度优于 O(n2) 的解决方案。

示例 1:

复制代码
输入:matrix = [[1,5,9],[10,11,13],[12,13,15]], k = 8
输出:13
解释:矩阵中的元素为 [1,5,9,10,11,12,13,13,15],第 8 小元素是 13

示例 2:

复制代码
输入:matrix = [[-5]], k = 1
输出:-5

这道题可以直接把二维数组变成一维数组然后排序

也可以选择 大根堆:

java 复制代码
class Solution {
    public int kthSmallest(int[][] matrix, int k) {

        int n=matrix.length;
        PriorityQueue<Integer> priorityQueue=new PriorityQueue<>(new Comparator<Integer>() {

            @Override
            public int compare(Integer o1, Integer o2) {
                return o2-o1;
            }
        });
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                if(priorityQueue.size()<k){
                    priorityQueue.add(matrix[i][j]);
                }else{
                    if(matrix[i][j]<priorityQueue.peek()){
                        priorityQueue.remove();
                        priorityQueue.add(matrix[i][j]);
                    }
                }

            }
        }
        return priorityQueue.peek();

    }
}
相关推荐
叶子爱分享3 分钟前
经典排序算法之归并排序(Merge Sort)
算法·排序算法
珹洺9 分钟前
C++算法竞赛篇:DevC++ 如何进行debug调试
java·c++·算法
SHUIPING_YANG16 分钟前
根据用户id自动切换表查询
java·服务器·数据库
Norvyn_719 分钟前
LeetCode|Day18|20. 有效的括号|Python刷题笔记
笔记·python·leetcode
爱吃烤鸡翅的酸菜鱼29 分钟前
IDEA高效开发:Database Navigator插件安装与核心使用指南
java·开发语言·数据库·编辑器·intellij-idea·database
惊涛骇浪、35 分钟前
SpringMVC + Tomcat10
java·tomcat·springmvc
呆呆的小鳄鱼1 小时前
leetcode:冗余连接 II[并查集检查环][节点入度]
算法·leetcode·职场和发展
墨染点香1 小时前
LeetCode Hot100【6. Z 字形变换】
java·算法·leetcode
沧澜sincerely1 小时前
排序【各种题型+对应LeetCode习题练习】
算法·leetcode·排序算法
CQ_07121 小时前
自学力扣:最长连续序列
数据结构·算法·leetcode