LeetCode每日一题 378

给定一个 n x n 矩阵,其中每行和每列元素均按升序排序,找到矩阵中第k小的元素。
请注意,它是排序后的第k小元素,而不是第k个元素。
思路:
直接变成一维排序即可

#!/user/bin/env python
# !-*-coding:utf-8 -*-
# !Time :2018/10/8 8:06 PM
# !Author : hyCong
# !@File  : .py
def kthSmallest(matrix, k):
    temp = []
    length = len(matrix[0])
    for i in range(length):
        for j in range(length):
            temp.append(matrix[i][j])

    temp = sorted(temp)
    return temp[k-1]


matrix = [
    [1, 5, 9],
    [10, 11, 13],
    [12, 13, 15]
]
print kthSmallest(matrix,8)

猜你喜欢

转载自blog.csdn.net/c630565685/article/details/82973679