给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第k小的元素。
请注意,它是排序后的第k小元素,而不是第k个元素。
示例:
matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,
返回 13。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution(object):
def kthSmallest(self, matrix, k):
"""
:type matrix: List[List[int]]
:type k: int
:rtype: int
"""
left = matrix[0][0]
right = matrix[-1][-1]
while left < right:
mid = (left+right) // 2
print('mid:%d' %(mid))
num = sum(bisect.bisect_right(row , mid) for row in matrix)
#bisect.bisect_right(row, mid)计算row中元素值<=mid的数量
print('num:%d' %(num))
if num < k:
left = mid+1
else:
right = mid
return left