Leetcode 1380：矩阵中的幸运数（超详细的解法！！！）

给你一个 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
• 矩阵中的所有元素都是不同的

解题思路

不难通过暴力来解，具体思路就是先找到每一行的最小值并且记录其位置，然后再判断该值是不是该列上的最大值。

class Solution:
    def luckyNumbers (self, A: List[List[int]]) -> List[int]:
        res = []
        r, c = len(A), len(A[0])
        for i, v in enumerate(A):
            t = (float("inf"), -1)
            for j in range(c):
                if A[i][j] < t[0]: t = (A[i][j], j)
                    
            for j in range(r):
                if A[j][t[1]] > t[0]: break
            else:
                res.append(t[0])
        return res

在评论区看到一个非常棒的解法，找到每一行的最小值建立一个集合，找到每一列的最大值建立一个集合，然后取二者交集（矩阵中的所有元素都是不同的）。

class Solution:
    def luckyNumbers(self, A: List[List[int]]) -> List[int]:
        return list({min(row) for row in A} & {max(col) for col in zip(*A)})

reference:

https://leetcode.com/problems/lucky-numbers-in-a-matrix/discuss/539751/Python-One-Liner-using-Set-Intersection

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题，希望大家指出！！！