零矩阵.org

题目

思路

遍历矩阵，找出0的行列索引，分别存进set()中，再遍历一遍矩阵，将行列在set中的元素置0

code

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        # 暴力法
        m, n = len(matrix), len(matrix[0])
        row, column = set(), set()
        for i in range(m):
            for j in range(n):
                if matrix[i][j] == 0:
                    row.add(i)
                    column.add(j)
        for i in range(m):
            for j in range(n):
                if i in row or j in column:
                    matrix[i][j] = 0