LeetCode-Python-1198. 找出所有行中最小公共元素(哈希表 + 矩阵降维)

给你一个矩阵 mat,其中每一行的元素都已经按 递增 顺序排好了。请你帮忙找出在所有这些行中 最小的公共元素。

如果矩阵中没有这样的公共元素,就请返回 -1。

 

示例:

输入:mat = [[1,2,3,4,5],[2,4,5,8,10],[3,5,7,9,11],[1,3,5,7,9]]
输出:5
 

提示:

1 <= mat.length, mat[i].length <= 500
1 <= mat[i][j] <= 10^4
mat[i] 已按递增顺序排列。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-smallest-common-element-in-all-rows
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:

已知每一行递增,所以不会有重复,

所以最小的出现频率为 len(mat)的元素就是答案。

学会了骚操作:

flatten = sum(mat, [])可以把二维矩阵展开到一维。原理目前不会。

时间复杂度:O(MN)

空间复杂度:O(MN)

class Solution(object):
    def smallestCommonElement(self, mat):
        """
        :type mat: List[List[int]]
        :rtype: int
        """
        from collections import Counter
        flatten = sum(mat,[])  # 把二维矩阵展开成一维
        dic = Counter(flatten) # 统计所有元素频率
        for num in mat[0]: # 在第一行元素里找有没有出现频率等于mat行数的元素
            if dic[num] == len(mat): # 第一次满足条件的元素就是答案
                return num
        return -1

 

你可能感兴趣的:(Leetcode)