[LeetCode] 378. Kth Smallest Element in a Sorted Matrix @ python

一.题目:
找到一个有序二维数组的第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
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值