LeetCode 378. 有序矩阵中第 K 小的元素

题目描述

🔥 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 小的元素
}

🍏 点击查看 Java 题解

1

write your code here
本文作者:
本文链接: https://hgnulb.github.io/blog/2022/32635930
版权声明: 本博客所有文章除特别声明外，均采用 CC BY-NC-SA 4.0 许可协议，转载请注明出处！