用python写leetcode【11】 -- 有序矩阵中第K小的元素（378）

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

题目

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

示例:

matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,

返回 13。
说明:
你可以假设 k 的值永远是有效的, 1 ≤ k ≤ n2 。

思路

看到这个题和搜索二维数组一样，我总是本能的想用二叉树，但是仔细想想并不成功，对于寻找分位数这种还是要用二分法。即设置l，r两个变量采用二分法的办法，每一次数一数有几个数小于中心值，然后循环直到l=r，此时l就是矩阵中存在的第k小数值。

代码

       class Solution:
        def kthSmallest(self, matrix, k):
            """
            :type matrix: List[List[int]]
            :type k: int
            :rtype: int
            """
            l = matrix[0][0]
            r = matrix[-1][-1]
            while(l<r):
                m = l+(r-l)//2
                total=sum(bisect.bisect_right(row, m) for row in matrix)
                if total>=k:
                    r= m
                else:
                    l = m+1
            return l    

代码中使用了bisect模块，sum(bisect.bisect_right(row, m) for row in matrix)
其中bisect.bisect_right(row, m）在row中查找m，m存在时返回m右侧的位置，m不存在返回应该插入的位置，这整行代码的意思就是数每一行row中在m值左边的个数并累加。