python第k序元素查找_LeetCode有序矩阵中第K小的元素(Python语言)

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

题目

给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。

请注意，它是排序后的第 k 小元素，而不是第 k 个不同的元素。

示例：

matrix = [

[ 1, 5, 9],

[10, 11, 13],

[12, 13, 15]

],

k = 8,

返回 13。

提示：

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

解题思路

先审题，题目中给出的是二维数组，而问题需要求得的是二维数组中第 k 小的元素。在这里，最直接的方法就是将二维数组转换为一维数组，对于转换后的一维数组进行升序排序，那么此时的一维数组中的第 k 个元素就是所求的结果。代码大致如下：

# python

def kthSmallest(self, matrix: List[List[int]], k: int) -> int:

res = []

for row in matrix:

for ele in row:

res.append(ele)

res.sort()

return res[k-1]

在这里，代码并没有使用矩阵的特性。而是转换为一维数组进行求解。这个解法的时间复杂度为 O(n^2logn)，即是对 n x n 个数进行排序。

因为上面的解法是将矩阵转换为一维数组，并不是在矩阵的基础上解决问题。下面使用二分查找，来尝试以在矩阵的前