73. 矩阵置零
给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用原地算法。
示例 1:
输入: [ [1,1,1], [1,0,1], [1,1,1] ] 输出: [ [1,0,1], [0,0,0], [1,0,1] ]
示例 2:
输入: [ [0,1,2,0], [3,4,5,2], [1,3,1,5] ] 输出: [ [0,0,0,0], [0,4,5,0], [0,3,1,0] ]
进阶:
- 一个直接的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
- 一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
- 你能想出一个常数空间的解决方案吗?
解题思路:首先想到用m+n的额外空间存储需要置零的行和列,但是题目要求使用原地算法求解。因此,可以先记录第一行和第一列是否应该置零,然后使用第一行和第一列的空间存储子数组(去除第一行与第一列)对应的行与列是否置零,之后对子数组进行置零操作,最后将第一行与第一列进行置零操作。
Python3代码如下:
class Solution(object):
def setZeroes(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: None Do not return anything, modify matrix in-place instead.
"""
row_first = reduce(lambda x,y:x or matrix[0][y] == 0,range(len(matrix[0])),False)
col_first = reduce(lambda x,y:x or matrix[y][0] == 0,range(len(matrix)),False)
for i in range(1,len(matrix)):
for j in range(1,len(matrix[0])):
if matrix[i][j] == 0:
matrix[i][0],matrix[0][j] = 0,0
for i in range(1,len(matrix)):
for j in range(1,len(matrix[0])):
if matrix[i][0] == 0 or matrix[0][j] == 0:
matrix[i][j] = 0
if row_first:
for j in range(len(matrix[0])):
matrix[0][j] = 0
if col_first:
for i in range(len(matrix)):
matrix[i][0] = 0