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

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

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

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

返回 13。

思路：
方法一：暴力法，把矩阵的所有元素排序，找出target元素，时间复杂度O（n*n）。
方法二：结合二分法
1.最小值left 在左下角，最大值在右下角，目标值 target 在[left,right]。 mid = left + (mid - left)//2 ,matrix 小于等于 mid 的数量
为 count， if count < k ,则target一定在mid 的右边，即 left = mid +1 ；if count >=k,则 target 一定在mid 的左边，包括mid ，即 right = mid。直到上下界 left == right ，返回上界/下界，即保证查到的值在矩阵内且小于等于该值的矩阵元素恰好有k个。

class Solution(object):
    def kthSmallest(self, matrix, k):

        """
        :type matrix: List[List[int]]
        :type k: int
        :rtype: int
        """
        row = len(matrix)
        col = len(matrix[0])
        left = matrix[0][0]
        right = matrix[-1][-1]
        while left < right:
            mid = left + (right - left)//2
            count = self.lessCount(matrix,mid)
            if k <=count: # (mid 之前小于mid的个数小于K) target 在mid 左边
                right = mid 
            elif k >count:#(mid 之前的小于mid 的个数 大于K) target 在mid 右边
                left = mid + 1 
        return left 
    def lessCount(self,matrix,mid): #统计小于等于 mid 的个数
        count = 0 
        row = len(matrix)
        col = len(matrix[0])
        i = row -1 
        j = 0 
        while i >=0 and j < col:
            if matrix[i][j] <= mid:
                count += i + 1
                j +=1 
            else:
                i -=1 
        print(count)
        return count