leetcode(76)_1380_easy_矩阵中的幸运数_python

矩阵中的幸运数

题目描述：
给你一个 m * n 的矩阵，矩阵中的数字 各不相同 。请你按 任意 顺序返回矩阵中的所有幸运数。
幸运数是指矩阵中满足同时下列两个条件的元素：
在同一行的所有元素中最小
在同一列的所有元素中最大
示例 ：
输入：matrix = [[3,7,8],[9,11,13],[15,16,17]]
输出：[15]
解释：15 是唯一的幸运数，因为它是其所在行中的最小值，也是所在列中的最大值。
提示：

• m == mat.length
• n == mat[i].length
• 1 <= n, m <= 50
• 1 <= matrix[i][j] <= 10^5
• 矩阵中的所有元素都是不同的

解法

模拟过程，先找各行最小值、各列最大值，然后逐个判断各元素是否是最小最大值。

代码

class Solution:
    def luckyNumbers (self, matrix: List[List[int]]) -> List[int]:
        row_min = [100001 for _ in range(len(matrix))]
        col_max = [0 for _ in range(len(matrix[0]))]
        for i in range(len(matrix[0])):
            for j in range(len(matrix)):
                row_min[j] = min(matrix[j][i], row_min[j])
                col_max[i] = max(matrix[j][i], col_max[i])
        res = []
        for i in range(len(matrix[0])):
            for j in range(len(matrix)):
                if row_min[j] == matrix[j][i] and col_max[i] == matrix[j][i]:
                    res.append(matrix[j][i])
        return res         

测试结果

执行用时：56 ms, 在所有 Python3 提交中击败了 31.52% 的用户
内存消耗：15.2 MB, 在所有 Python3 提交中击败了 38.13% 的用户

说明

算法题来源：力扣(LeetCode)