LeetCode：73. Set Matrix Zeroes - Python

问题描述：

73. 矩阵置零

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

示例 :

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

进阶:

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

问题分析：

题目提示已经给出了，想必大家一定可以想到前两种方法，现在介绍另外一种方法，就不开辟空间的方法。

（1）要设置含有 0 的整列和整行全为 0，可以扫描一遍矩阵，在原始矩阵的第一列和第一行，做标记，即，这一行或者这一列是否要置为 0。

（2）最后根据第一列和第一行的标志，对整列、行，进行置 0 处理。

（3）需要注意的是，要特别注意第一行和第一列含有 0 的情况，不能和标志混淆。

Python3实现：

class Solution:
    def setZeroes(self, matrix):
        
        first_col = False  # 用于记录 第一列是否有 0 没有为false

        n, m = len(matrix), len(matrix[0])

        for i in range(n):
            if matrix[i][0] == 0:  # 检测第一列是否有 0
                first_col = True
            for j in range(1, m):  # 从第二列开始，一行一行的扫描，然后在第一列和第一行上进行标记。
                if matrix[i][j] == 0:
                    matrix[0][j] = matrix[i][0] = 0

        for i in range(1, n):  # 根据第一列和第一行的标记，对整个矩阵置0处理，不包括第一行和第一列。
            for j in range(1, m):
                if matrix[i][0] == 0 or matrix[0][j] == 0:
                    matrix[i][j] = 0

        if matrix[0][0] == 0:  # 处理第一行
            for j in range(m):
                matrix[0][j] = 0

        if first_col:  # 处理第一列
            for i in range(n):
                matrix[i][0] = 0


if __name__ == '__main__':
    matrix = [
        [1, 1, 1],
        [1, 0, 1],
        [1, 1, 1]]
    solu = Solution()
    solu.setZeroes(matrix)
    for row in matrix:
        print(row)

声明： 总结学习，有问题或不妥之处，可以批评指正哦。

题目链接：leetcode-cn.com/problems/set-matrix-zeroes/ ，参考官方解答