给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第 k 小的元素。
请注意,它是排序后的第 k 小元素,而不是第 k 个不同的元素。
找规律
这是力扣网站的一道题,题目给了一个例子
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8
我们把它拆分成这样几个矩阵
- [1]
- [ [ 1, 5]
[10, 11] ] - [ [ 1, 5, 9],
[10, 11, 13],
[12, 13, 15] ]
我们会发现每个矩阵最后一个元素会是这个矩阵里值最大的元素,我们要找第8小的元素,第一个矩阵只有1个元素,第二个矩阵有4个元素,所以不在这两个矩阵里,第三个矩阵有9个元素,所以我们要找的肯定是图中标亮的几个元素的一个
查找
规律找到了以后,接下来要解决的就是怎么查找了
如果我们用暴力查找的话,也可以做出来,但是时间复杂度会比较大,所以我们思考还有没有别的更加快的查找方式
我们知道,二分查找是效率比较高的查找方式,但它需要数组有序,我们观察题目,会发现其实这个二维数组其实是相对有序的(其实我觉得题目给的例子还是比较特殊的,这个等会再说)
这里用二分查找的思路是:找到这个二维数组的中间值,(不是中间索引号,这里跟一维数组的情况不太一样),然后求出这个中间值是第几小的元素,如果比k大,那么说明k指向的数在它的“前边”,否则在它的“后边”
var kthSmallest = function(matrix, k) {
let n = matrix.length;