python实现leetcode之73. 矩阵置零

解题思路

第一步，找出出现0的行列
第二步，对出现0的行清0
第三步，对出现0的列清0

73. 矩阵置零

代码

class Solution(object):
    def setZeroes(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: None Do not return anything, modify matrix in-place instead.
        """
        row_trace, col_trace = set(), set()
        # 第一步，找出出现0的行列
        for i in range(len(matrix)):
            for j in range(len(matrix[i])):
                if matrix[i][j] == 0:
                    row_trace.add(i)
                    col_trace.add(j)
        # 第二步，对出现0的行清0
        for i in row_trace:
            for j in range(len(matrix[i])):
                matrix[i][j] = 0
        # 第三步，对出现0的列清0
        for j in col_trace:
            for i in range(len(matrix)):
                matrix[i][j] = 0