73. 矩阵置零
题目
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
示例
示例 1:
示例 2:
提示:
进阶:
1. 两个set存储横纵坐标
执行用时:28 ms, 在所有 Python 提交中击败了69.77%的用户
内存消耗:13.5 MB, 在所有 Python 提交中击败了88.37%的用户
通过测试用例:170 / 170
class Solution(object):
def setZeroes(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: None Do not return anything, modify matrix in-place instead.
"""
lm,lmm = len(matrix), len(matrix[0])
# 两个set,分别装坐标x和y
sx = set()
sy = set()
# 遍历寻找需要为0的横纵
for i in range(lm):
for j in range(lmm):
if matrix[i][j] == 0:
sx.add(i)
sy.add(j)
# 部署零
for i in range(lm):
for j in range(lmm):
if i in sx or j in sy:
matrix[i][j] = 0