LeetCode 378. 有序矩阵中第 K 小的元素
题目描述
思路分析
大顶堆
二分查找
参考代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
func kthSmallest(matrix [][]int, k int) int {
n := len(matrix)
low, high := matrix[0][0], matrix[n-1][n-1]
for low < high {
mid := low + (high-low)/2 // 使用二分法确定当前的中间值
count := 0
j := n - 1
// 对每一行进行遍历，寻找小于等于 mid 的元素数量
for i := 0; i < n; i++ {
for j >= 0 && matrix[i][j] > mid {
j--
}
count += j + 1
}
// 根据小于等于 mid 的元素数量来调整 low 和 high
if count < k {
low = mid + 1
} else {
high = mid
}
}
return low // 最终返回的 low 即为第 k 小的元素
}
1
write your code here
相似题目
| 题目 | 难度 | 题解 |
|---|---|---|
| 查找和最小的 K 对数字 | Medium | |
| 乘法表中第 k 小的数 | Hard | |
| 找出第 K 小的数对距离 | Hard | |
| 第 K 个最小的素数分数 | Medium |
本博客所有文章除特别声明外，均采用
CC BY-NC-SA 4.0
许可协议，转载请注明出处！