Golang Leetcode 378. Kth Smallest Element in a Sorted Matrix.go

原创于 2019-04-08 14:55:58 发布 · 141 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#golang #leetcode #算法

leetcode-golang 专栏收录该内容

223 篇文章

订阅专栏

本文介绍了一种使用二分查找算法在有序矩阵中寻找第K小元素的方法。通过设定查找范围并计算小于等于中间值的元素数量，逐步缩小范围直至找到目标元素。该方法效率高，适用于大规模数据处理。

思路

二分查找，查找的同时计算个数

code

func kthSmallest(matrix [][]int, k int) int {
	if len(matrix) == 0 {
		return 0
	}
	m := len(matrix)
	low, high := matrix[0][0], matrix[m-1][m-1]
	for low < high {
		mid := (low + high) / 2
		cc := helper(matrix, mid)
		if cc < k {
			low = mid + 1
		} else {
			high = mid
		}
	}
	return low
}
func helper(matrix [][]int, target int) int {
	c := 0
	i, j := len(matrix)-1, 0
	for i >= 0 && j <= len(matrix)-1 {
		if matrix[i][j] <= target {
			j++
			c += i + 1
		} else {
			i--
		}
	}
	return c
}

更多内容请移步我的repo：https://github.com/anakin/golang-leetcode