LeetCode 热题100-18 矩阵置零

矩阵置零

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。

示例 1：

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]

示例 2：

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

提示：

• m == matrix.length
• n == matrix[0].length
• 1 <= m, n <= 200
• -231 <= matrix[i][j] <= 231 - 1

进阶：

• 一个直观的解决方案是使用  O(mn) 的额外空间，但这并不是一个好的解决方案。
• 一个简单的改进方案是使用 O(m + n) 的额外空间，但这仍然不是最好的解决方案。
• 你能想出一个仅使用常量空间的解决方案吗？

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        hashmap_j = {}
        hashmap_i = {}
        for i,iv in enumerate(matrix):
            for j,jv in enumerate(iv):
                if jv == 0 :
                    if j not in hashmap_j:
                        hashmap_j[j] = 1
                    if i not in hashmap_i:
                        hashmap_i[i] = 1
        for z in range(len(matrix[0])):
            for i in hashmap_i.keys():
                matrix[i][z] = 0
        for z in range(len(matrix)):
            for j in hashmap_j.keys():
                matrix[z][j] = 0

比较容易想的是o(m+n)的，这边用了两个hash表来存哪些行和列需要置零，然后从中取出来遍历着置为零即可。另外还想到一个想法是递归的方法，找到零的时候就往下执行下一行元素起始的递归，当前递归后再执行当前元素的行列置零操作。

然后看了一下其他的解法是o(1)的，既然是所在元素的行和列置零，那么第一行和第一列自然能够成为具有存储功能的空间。其实也就是找到了m[i][j]后，将m[i][0]与m[0][j]都赋值为0就好了，然后最后再根据这两个信息来执行更新操作。