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();

    }
}
相关推荐
星空下的曙光几秒前
Node.js crypto模块所有 API 详解 + 常用 API + 使用场景
算法·node.js·哈希算法
爱读源码的大都督24 分钟前
为什么有了HTTP,还需要gPRC?
java·后端·架构
Lucky_Turtle43 分钟前
【Java Xml】Apache Commons Digester3解析
xml·java·apache
聪明的笨猪猪1 小时前
Java Redis “缓存设计”面试清单(含超通俗生活案例与深度理解)
java·经验分享·笔记·面试
FIavor.1 小时前
我发送给Apifox是http://localhost:9002/goods/getByUserName?name=张三 为什么会是500哪里错了?
java·服务器·网络协议·http
ID_180079054731 小时前
京东获取整站实时商品详情数据|商品标题|数据分析提取教程
java·开发语言
StarPrayers.2 小时前
旅行商问题(TSP)(2)(heuristics.py)(TSP 的两种贪心启发式算法实现)
前端·人工智能·python·算法·pycharm·启发式算法
微露清风2 小时前
系统性学习C++-第五讲-内存管理
java·c++·学习
计算机毕业设计木哥2 小时前
计算机毕业设计选题推荐:基于SpringBoot和Vue的快递物流仓库管理系统【源码+文档+调试】
java·vue.js·spring boot·后端·课程设计
爱吃橘的橘猫2 小时前
嵌入式系统与嵌入式 C 语言(2)
c语言·算法·嵌入式