题目描述：

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

 

示例 1：

 

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]

示例 2：

 

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

 

提示：

• m == matrix.length
• n == matrix[0].length
• 1 <= m, n <= 200
• -231 <= matrix[i][j] <= 231 - 1

 

 

我的作答：

犯了一个逻辑错误，当遍历到0元素时我直接给行和列置零了。。但是这样会导致后面再遍历，每置零的元素的行和列会再置零。。导致0多了

所以思路是先储存0元素的索引，再分别置零行和列。由于行和列有可能会重合，所以使用集合来储存行和列索引

python">class Solution(object):def setZeroes(self, matrix):""":type matrix: List[List[int]]:rtype: None Do not return anything, modify matrix in-place instead."""if not matrix: matrix = [[]]m = len(matrix)n = len(matrix[0])row = set()col = set()for i in range(m):for j in range(n):if matrix[i][j]==0:row.add(i) #行col.add(j) #列for i in row:matrix[i] = [0]*nfor j in col:for row in range(m):matrix[row][j] = 0

 

参考：

python">class Solution(object):def setZeroes(self, matrix):""":type matrix: List[List[int]]:rtype: None Do not return anything, modify matrix in-place instead."""row = len(matrix)col = len(matrix[0])row_zero = set()col_zero = set()for i in range(row):for j in range(col):if matrix[i][j] == 0:row_zero.add(i)col_zero.add(j)for i in range(row):for j in range(col):if i in row_zero or j in col_zero:matrix[i][j] = 0