博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
leetcode378. Kth Smallest Element in a Sorted Matrix
阅读量:5924 次
发布时间:2019-06-19

本文共 2001 字,大约阅读时间需要 6 分钟。

题目要求

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.

在一个从左到右,从上到下均有序的二维数组中,找到从小到第k个数字,这里需要注意,不要求一定要是唯一的值,即假设存在这样一个序列1,2,2,3,则第三个数字是2而不是3。

思路一:优先队列

当涉及到从一个集合中查找一个元素这样的问题时,我们往往会立刻想到查找的几种方式:有序数组查找,无序数组查找,堆排序。这里如果将二维数组转化为一维有序数组,成本未免太大了。同理,将其中所有元素都转化为堆,也会存在内存不足的问题。因此我们可以采用部分元素堆排序即可。即我们每次只需要可能构成第k个元素的值进行堆排序就可以了。

public int kthSmallest(int[][] matrix, int k) {        //优先队列        PriorityQueue
queue = new PriorityQueue
(); //将每一行的第一个元素放入优先队列中 for(int i = 0 ; i
{ int x; int y; int value; public Tuple(int x, int y, int value) { this.x = x; this.y = y; this.value = value; } @Override public int compareTo(Tuple o) { // TODO Auto-generated method stub return this.value - o.value; } }

思路二:二分法查找

二分查找的核心问题在于,如何找到查找的上界和下届。这边我们可以矩阵中的最大值和最小值作为上界和下界。然后不停的与中间值进行比较,判断当前矩阵中小于该中间值的元素有几个,如果数量不足k,就将左指针右移,否则,就将右指针左移。直到左右指针相遇。这里需要注意,不能在数量等于k的时候就返回mid值,因为mid值不一定在矩阵中存在。

public int kthSmallest2(int[][] matrix, int k){        int low = matrix[0][0], high = matrix[matrix.length-1][matrix[0].length-1];        while(low <= high) {            int mid = low + (high - low) / 2;            int count = 0;            int i = matrix.length-1 , j = 0;            //自矩阵左下角开始计算比mid小的数字的个数            while(i>=0 && j < matrix.length){                if(matrix[i][j]>mid) i--;                else{                    count+=i+1;                    j++;                }            }            if(count < k) {                low = mid + 1;            }else{                high = mid - 1;            }        }        return low;    }

转载地址:http://dmavx.baihongyu.com/

你可能感兴趣的文章
Mobx 与 Redux 的性能对比
查看>>
vue-cli 3.0配置webpack目录别名alias
查看>>
web布局固定宽度+变化宽度实现思路
查看>>
微信小程序黑客马拉松即将开始,来做最酷的 Mini Program Creators!
查看>>
前端工程化:围绕Jenkins打造工作流的过程
查看>>
前端技术周刊 2018-12-03:DOM
查看>>
搭建 vue2 单元测试环境(karma+mocha+webpack3)
查看>>
Unity 游戏框架搭建 (九) 减少加班利器-QConsole
查看>>
Safari 版本回退方法
查看>>
Android 4 +https(如何启动TLS1 1 and TLS1 2)
查看>>
从shiro源码角度学习工厂方法设计模式
查看>>
python面试题~反射,元类,单例
查看>>
java多线程编程——锁优化
查看>>
写一个易于维护使用方便性能可靠的Hybrid框架(一)—— 思路构建
查看>>
为什么阿里巴巴禁止把SimpleDateFormat定义为static类型的?
查看>>
ResourceManager中的Resource Estimator框架介绍与算法剖析
查看>>
5分钟内看懂机器学习和深度学习的区别
查看>>
使用Network Recycle Bin启用映射网络驱动器上的回收站
查看>>
量子计算机的现状和趋势
查看>>
iOS - block变量捕获原理
查看>>