LeetCode|378.有序矩阵中第k小的元素

LeetCode习题答案汇总

题目：
给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。
请注意，它是排序后的第 k 小元素，而不是第 k 个不同的元素。

示例：

matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,

返回 13。

提示：
你可以假设 k 的值永远是有效的，1 ≤ k ≤ n2 。

分析：

• 将二维数组另存为为一维数组
• 对该一维数组进行排序
• 最后这个一维数组中的第 k个元素为答案。

代码：

class Solution:
    def kthSmallest(self, matrix: List[List[int]], k: int) -> int:
        res = [col for row in matrix for col in row]
        res.sort()
        return res[k-1]

结果：