每日一道Leetcode - 378. 有序矩阵中第 K 小的元素【堆｜归并排序】

最新推荐文章于 2021-02-26 19:45:23 发布

奋进的杰西

最新推荐文章于 2021-02-26 19:45:23 发布

阅读量93

点赞数

分类专栏： Leetcode 文章标签： python

本文链接：https://blog.csdn.net/weixin_41041275/article/details/113841791

版权

Leetcode 专栏收录该内容

304 篇文章 1 订阅

订阅专栏

在这里插入图片描述

class Solution:
    def kthSmallest(self, matrix: List[List[int]], k: int) -> int:
        n = len(matrix)
        # 存储每行第一个值，以及其对应下标
        queue = [(matrix[i][0],i,0) for i in range(n)]
        # 创建堆，直接一个列表所有元素入堆
        heapq.heapify(queue)

        res = []
        for i in range(k):
            num,x,y = heapq.heappop(queue)
            res.append(num)
            if y!=n-1:
                # 没有达到边界的情况
                # 将当前弹出元素的右边的元素继续入栈
                heapq.heappush(queue,(matrix[x][y+1],x,y+1))
        return res[-1]