问题描述:
给定一个m x n
的矩阵,如果一个元素为 0
,则将其所在行和列的所有元素都设为 0
。请使用原地算法。
示例 :
输入:
[ [1,1,1],
[1,0,1],
[1,1,1]]
输出:
[ [1,0,1],
[0,0,0],
[1,0,1]]
进阶:
- 一个直接的解决方案是使用
O(mn)
的额外空间,但这并不是一个好的解决方案。 - 一个简单的改进方案是使用
O(m + n)
的额外空间,但这仍然不是最好的解决方案。 - 你能想出一个常数空间的解决方案吗?
问题分析:
题目提示已经给出了,想必大家一定可以想到前两种方法,现在介绍另外一种方法,就不开辟空间的方法。
(1)要设置含有 0
的整列和整行全为 0
,可以扫描一遍矩阵,在原始矩阵的第一列和第一行,做标记,即,这一行或者这一列是否要置为 0
。
(2)最后根据第一列和第一行的标志,对整列、行,进行置 0
处理。
(3)需要注意的是,要特别注意第一行和第一列含有 0
的情况,不能和标志混淆。
Python3实现:
class Solution:
def setZeroes(self, matrix):
first_col = False # 用于记录 第一列是否有 0 没有为false
n, m = len(matrix), len(matrix[0])
for i in range(n):
if matrix[i][0] == 0: # 检测第一列是否有 0
first_col = True
for j in range(1, m): # 从第二列开始,一行一行的扫描,然后在第一列和第一行上进行标记。
if matrix[i][j] == 0:
matrix[0][j] = matrix[i][0] = 0
for i in range(1, n): # 根据第一列和第一行的标记,对整个矩阵置0处理,不包括第一行和第一列。
for j in range(1, m):
if matrix[i][0] == 0 or matrix[0][j] == 0:
matrix[i][j] = 0
if matrix[0][0] == 0: # 处理第一行
for j in range(m):
matrix[0][j] = 0
if first_col: # 处理第一列
for i in range(n):
matrix[i][0] = 0
if __name__ == '__main__':
matrix = [
[1, 1, 1],
[1, 0, 1],
[1, 1, 1]]
solu = Solution()
solu.setZeroes(matrix)
for row in matrix:
print(row)
声明: 总结学习,有问题或不妥之处,可以批评指正哦。
题目链接:leetcode-cn.com/problems/set-matrix-zeroes/ ,参考官方解答