给定一个 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的矩阵,按照题目要求把对应该是0的部分变成0,其他照抄,最后复制回来,这种比较简单,没有实现。
第二种稍微聪明一点的思路是新建两个数组,长度分别为m,n,其中每个元素如果有标记,就分别代表对应的行或者列都应该是0,然后线性扫描输入的二维数组,如果当前元素是0,就把对应的行数和列数在m,n数组中标记一下,如果不是0,就查询m, n 数组,看看当前行或者当前列是不是应该归0,如果是,就把它变成0,如果不是就处理下一个元素。
class Solution(object):
def setZeroes(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: void Do not return anything, modify matrix in-place instead.
"""
self.matrix = matrix
m = len(matrix)
n = len(matrix[0])
l_m = [1] * m
l_n = [1] * n
for i in range(m):
for j in range(n):
# print matrix[i][j], i, j
if matrix[i][j] == 0:
l_m[i] = 0
l_n[j] = 0
# print l_m, l_n
for i in range(m):
for j in range(n):
# print i,j
if l_m[i] == 0 or l_n[j] == 0:
matrix[i][j] = 0
# print matrix
第三种思路原地算法:
把需要变0的元素暂时变成“0”
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 or not matrix[0]:
return matrix
m, n = len(matrix), len(matrix[0])
for i in range(m):
for j in range(n):
if matrix[i][j] == 0:
for t in range(m):#同一列
if matrix[t][j] != 0:
matrix[t][j] = "0"
for t in range(n):#同一行
if matrix[i][t] != 0:
matrix[i][t] = "0"
for i in range(m):
for j in range(n):
if matrix[i][j] == "0":
matrix[i][j] = 0