[LeetCode] 378. Kth Smallest Element in a Sorted Matrix @ python_leetcode378完整代码

一．题目：
找到一个有序二维数组的第k小的数字。这个二维数组，每行依次增加，每列依次增加。
Example:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
return 13.

二．解题思路：
首先笨办法就是将二维数组的数据放入到对里面排序，取其中第k小的元素值返回，居然通过了．
代码如下：

class Solution(object):
    def kthSmallest(self, matrix, k):
        """
        :type matrix: List[List[int]]
        :type k: int
        :rtype: int
        """
        num_lists = []
        heapq.heapify(num_lists)
        for i in matrix:
            for j in i:
                heapq.heappush(num_lists,j)
        return heapq.nsmallest(k,num_lists)[-1]

看网上更优的解法是利用二分搜索，将查找上下限设为矩阵的右下角与左上角元素，查找过程中对中值在矩阵每一行的位置进行累加，记该值为loc，再根据loc与k的大小关系调整上下限．
代码如下：

class Solution(object):
    def kthSmallest(self, matrix, k):
        """
        :type matrix: List[List[int]]
        :type k: int
        :rtype: int
        """
        lo, hi = matrix[0][0], matrix[-1][-1]
        while lo <= hi:
            mid = (lo + hi) >> 1
            loc = sum(bisect.bisect_right(m, mid) for m in matrix)
            if loc >= k:
                hi = mid - 1
            else:
                lo = mid + 1
        return lo