力扣 378. 有序矩阵中第K小的元素

最新推荐文章于 2023-01-11 22:29:00 发布

csu_xiji

最新推荐文章于 2023-01-11 22:29:00 发布

阅读量178

点赞数

分类专栏：力扣二分

本文链接：https://blog.csdn.net/xiji333/article/details/109140798

版权

力扣同时被 2 个专栏收录

282 篇文章 2 订阅

订阅专栏

二分

54 篇文章 1 订阅

订阅专栏

https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix/
在这里插入图片描述
思路：显然易得矩阵最小值为 $m [0] [0]$ ，最大值为 $m [n - 1] [n - 1]$ ，那么我们可以二分第 $k$ 小元素的值 $m i d$ ，再遍历矩阵的每一行(或列)判断 $< = m i d$ 的数的总和，假设为 $n u m$ ，如果 $n u m > = k$ 说明 $m i d$ 过大，否则说明 $m i d$ 过小。

class Solution:
    def upper_bound(self,array,k):
        l,r=0,len(array)-1
        while l<=r:
            mid=(l+r)//2
            if array[mid]>k:
                r=mid-1
            else:
                l=mid+1
        return l

    def kthSmallest(self, matrix: List[List[int]], k: int) -> int:
        n=len(matrix)-1
        l,r=matrix[0][0],matrix[n][n]
        while l<=r:
            mid=(l+r)//2
            num=0
            for each in matrix:
                if num>=k:
                    break
                num+=self.upper_bound(each,mid)
            if num>=k:
                r=mid-1
            else:
                l=mid+1
        return l