题目分析:
这一题是给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。并且让我们在原数组上做修改,我们可以先遍历一遍,用set()纪录0所在的行列。再用两个for把对应行列元素至为0。
测试代码: (Runtime: 92 ms, faster than 75.68% )
class Solution:
def setZeroes(self, matrix: list) -> None:
h = set()
l = set()
for i in range(len(matrix)):
for j in range(len(matrix[0])):
if matrix[i][j] == 0:
h.add(i)
l.add(j)
for i in h:
# matrix[i] = [0] * len(matrix[0])
for j in range(len(matrix[0])):
matrix[i][j] = 0
for j in l:
for i in range(len(matrix)):
matrix[i][j] = 0
return matrix #注意,返回值只是为了方便查看结构,题目不要返回,提交时请删除该行
print(Solution().setZeroes([[0,1,2,0],[3,4,5,2],[1,3,1,5]])) #提交时请删除该行