题目:
给你一个 m * n 的矩阵,矩阵中的数字 各不相同 。请你按 任意 顺序返回矩阵中的所有幸运数。
幸运数是指矩阵中满足同时下列两个条件的元素:
在同一行的所有元素中最小
在同一列的所有元素中最大
示例 1:
输入:matrix = [[3,7,8],[9,11,13],[15,16,17]]
输出:[15]
解释:15 是唯一的幸运数,因为它是其所在行中的最小值,也是所在列中的最大值。
示例 2:
输入:matrix = [[1,10,4,2],[9,3,8,7],[15,16,17,12]]
输出:[12]
解释:12 是唯一的幸运数,因为它是其所在行中的最小值,也是所在列中的最大值。
示例 3:
输入:matrix = [[7,8],[1,2]]
输出:[7]
提示:
m == mat.length
n == mat[i].length
1 <= n, m <= 50
1 <= matrix[i][j] <= 10^5
矩阵中的所有元素都是不同的
分析:
- min_row表示行最小
- max_col表示列最大
- 通过遍历matrix矩阵找到这个幸运数即可
代码:
class Solution:
def luckyNumbers (self, matrix: List[List[int]]) -> List[int]:
n = len(matrix) # 行
m = len(matrix[0]) # 列
min_row = [10**6 for i in range(n)]
max_col = [0 for i in range(m)]
for i in range(n):
for j in range(m):
min_row[i] = min(min_row[i],matrix[i][j])
max_col[j] = max(max_col[j],matrix[i][j])
return [x for x in min_row if x in max_col]
结果: