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

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

原始题目链接：https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix/

给你一个 n x n 矩阵 matrix ，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。
请注意，它是 排序后 的第 k 小元素，而不是第 k 个 不同 的元素。

你必须找到一个内存复杂度优于 O(n2) 的解决方案。

示例 1：

输入：matrix = [[1,5,9],[10,11,13],[12,13,15]], k = 8
输出：13
解释：矩阵中的元素为 [1,5,9,10,11,12,13,13,15]，第 8 小元素是 13
示例 2：

输入：matrix = [[-5]], k = 1
输出：-5

解题思路：

使用堆或是其他排序方法，是比较容易解题的，但时间复杂度高，并且没有利用题目的条件性质，看到有序，自然会想到使用二分查找的方式，这道题需要想清楚，每次二分查找的时候，需要哪两个变量进行比较。这里比较的两个变量是，第k小的元素和目前从左下角到右上角行走的路线统计了多少个小于k的元素的个数之间的比较。从左下角的路线开始查找也是不好想到的一个方法，具体实现看代码及注释。时间复杂度：O(nlog(r-l))，二分查找进行次数为 O(log(r-l))，每次操作时间复杂度为 O(n)。空间复杂度：O(1)

代码实现：

class Solution:
    def kthSmallest(self, matrix: List[List[int]], k: int) -> int:
        # 二维数组中以维数组的个数，即二维数组的行的维度
        row = len(matrix)

        def helper(mid):
            # 设置两个下标来从左下角开始遍历二维数组去统计比第k小元素的个数有多少
            i, j = row - 1, 0
            num = 0
            while i >= 0 and j < row:
                if matrix[i][j] <= mid:
                    # 当前元素matrix[i][j]是小于中位数的，则当前元素本身及其它所在的这一列都是小于mid的
                    # 所以num加上这一列的个数
                    num += i + 1
                    # 同时向右边移动，以为根据题目的条件，从左向右边是递增的，需要利用好这个性质
                    j += 1
                else:
                    # 这个情况是，当前元素matrix[i][j]是大于中位数的，所以根据题目的条件，需要向上移动
                    # 因为从下到上是递减的
                    i -= 1

            # 返回num(元素较小部分的元素个数)与k之间的比较结果，用于二分查找的条件判断
            return num >= k

        # 二分查找
        # 二维数组的从左到右和从上到下是递增的
        # 所以最小值设置为二维数组的第一个值
        left = matrix[0][0]
        # 最大值设置为二维数组的最后一个值
        right = matrix[-1][-1]
        
        while left < right:
            # 中位数的计算，右移表示除2
            # mid的计算需要放在while循环中
            mid = (left + right) >> 1

            # 每次将元素较小部分的个数和k做比较来进行二分查找
            # 需要一个计算目前有多少个较小部分的个数(和第k小元素比较)的辅助函数
            if helper(mid):
                # 元素较小部分的元素个数多了，需要减少一些，就是缩小查找范围
                # 就让右边界缩小
                right = mid
            else:
                # 否则元素个数少了，让左边界大一点
                left = mid + 1
        
        return left

参考文献：
https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix/solution/ci-ti-er-fen-cha-zhao-guo-cheng-de-dong-tu-yan-shi/