每日一题73-矩阵置零

1.题目详情

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

2.解题思路

因为只能当前数组中改，而每因为矩阵中某个值为0将所在行列变为0会改变整个矩阵的状态，因此遍历时需要记住所在行列为0的索引。

3.代码实现

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        m = len(matrix) # 行数
        n = len(matrix[0]) # 列数
        r,l = [],[] # 行列分别对应列表
        for i in range(m):
            for j in range(n):
                if matrix[i][j] == 0:
                    r.append(i) # 原列表中的行或列含0存入列表中
                    l.append(j)
        for o in range(len(r)): # 行中含0的那行变为0
            for h in range(n):
                matrix[r[o]][h] = 0
        for u in range(len(l)): # 列中含0的那列变为0
            for g in range(m):
                matrix[g][l[u]] = 0

4.知识点

–