LeetCode-Python-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
  • 矩阵中的所有元素都是不同的

思路:

找每行最小,和每列最大,最后取交集返回即可。

时间复杂度:O(MN)

空间复杂度:O(M + N)

知识点:zip(*matrix) 可以转置矩阵,非常方便取列

class Solution(object):
    def luckyNumbers (self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: List[int]
        """
        row, col = set(), set()
        
        for r in matrix:
            row.add(min(r))

        for c in zip(*matrix):
            col.add(max(c))
        
        return list(row & col)

 

发布了722 篇原创文章 · 获赞 105 · 访问量 19万+

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/104981546