LeetCode-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 。


class Solution:
    def kthSmallest(self, matrix, k):
        nums = []
        len1 = len(matrix)
        len2 = len(matrix[0])
        for i in range(len1):
            for j in range(len2):
                nums.append(matrix[i][j])
        nums.sort()
        return nums[k-1]