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

使用二分查找解决LeetCode378：有序矩阵中第K小的元素

最新推荐文章于 2023-05-02 16:39:56 发布

原创最新推荐文章于 2023-05-02 16:39:56 发布 · 556 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数据结构

leetcode刷题专栏收录该内容

252 篇文章

订阅专栏

该博客介绍了一种利用二分查找算法来解决LeetCode378问题的方法。通过从有序矩阵的左上角开始，逐步缩小搜索范围，找到第K小的元素。具体实现包括设置初始边界，计算小于中间值的元素个数，并根据比较结果调整搜索边界。代码中展示了详细的函数实现和辅助函数，用于计算小于特定值的元素数量。

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

1. 问题描述

在这里插入图片描述

2. 思路

左上角元素作为left，右下角元素作为right
执行二分查找

3. 代码

func kthSmallest(matrix [][]int, k int) int {
    n := len(matrix)
    left, right := matrix[0][0], matrix[n-1][n-1]   
    
    for left <= right {
        mid := left + (right - left) / 2
        // 计算matrix矩阵中小于mid 的元素个数
        smallNum := countSmall(matrix, mid, n)
        // fmt.Println(smallNum)
        if smallNum == k {
            right = mid - 1  // 不要怕错过，最终一定会在此处汇聚
        } else if smallNum < k {
            // mid太小，增大mid
            left = mid + 1
        } else {
            right = mid - 1
        }
    }
    return left
}

// 计算matrix矩阵中小于mid 的元素个数
func countSmall(matrix [][]int, mid int, n int) int {
    var num int
    i, j := n - 1, 0

    for i >= 0 && j <= n - 1 {
        if matrix[i][j] <= mid {
            j++
            num += i + 1
        } else {
            i--
        }
    }
    return num
}