LeetCode-Python-73. 矩阵置零

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

示例 1:

输入: 
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出: 
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

示例 2:

输入: 
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出: 
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

进阶:

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

 

思路：

第一种麻瓜思路是复制一个一模一样的M*N的矩阵，按照题目要求把对应该是0的部分变成0，其他照抄，最后复制回来，这种比较简单，没有实现。

第二种稍微聪明一点的思路是新建两个数组，长度分别为m，n，其中每个元素如果有标记，就分别代表对应的行或者列都应该是0，然后线性扫描输入的二维数组，如果当前元素是0，就把对应的行数和列数在m,n数组中标记一下，如果不是0，就查询m, n 数组，看看当前行或者当前列是不是应该归0，如果是，就把它变成0，如果不是就处理下一个元素。

class Solution(object):
    def setZeroes(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: void Do not return anything, modify matrix in-place instead.
        """
        self.matrix = matrix
        m = len(matrix)
        n = len(matrix[0])
        l_m = [1] * m
        l_n = [1] * n
        for i in range(m):
            for j in range(n):
                # print matrix[i][j], i, j
                if matrix[i][j] == 0:
                    l_m[i] = 0
                    l_n[j] = 0
                    
        # print l_m, l_n
        for i in range(m):
            for j in range(n):
                # print i,j
                if l_m[i] == 0 or l_n[j] == 0:
                        matrix[i][j] = 0
                # print matrix

第三种思路原地算法：

把需要变0的元素暂时变成“0”

class Solution(object):
    def setZeroes(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: None Do not return anything, modify matrix in-place instead.
        """
        if not matrix or not matrix[0]:
            return matrix
        m, n = len(matrix), len(matrix[0])
        
        for i in range(m):
            for j in range(n):
                if matrix[i][j] == 0:
                    for t in range(m):#同一列
                        if matrix[t][j] != 0:
                            matrix[t][j] = "0"
                    for t in range(n):#同一行
                        if matrix[i][t] != 0:
                            matrix[i][t] = "0"            
        for i in range(m):
            for j in range(n):
                if matrix[i][j] == "0":
                    matrix[i][j] = 0