给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 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.
"""
index = []
for i in range(len(matrix)):
for j in range(len(matrix[i])):
if matrix[i][j] == 0:
index.append((i, j))
for i, j in index:
matrix[i] = [0] * len(matrix[i])
for k in range(len(matrix)):
matrix[k][j] = 0
return