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