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: .pydef kthS...
·
给定一个 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)
更多推荐
所有评论(0)