零矩阵
概述:编写一种算法,若 M × N 矩阵中某个元素为 0 ,则将其所在的行与列清零。
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
输入:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
方法一:标记+循环
思路:首先二重循环,标记出原矩阵里面元素为 0 的位置。再对标记列表循环,依次把行和列全部替换成 0 。
# 标记+循环
class Solution:
def setZeroes(self, matrix: List[List[int]]) -> None:
m = len(matrix)
n = len(matrix[0])
ans = []
for i in range(m):
for j in range(n):
if matrix[i][j] == 0:
ans.append([i,j])
for i in ans:
row = i[0]
col = i[1]
matrix[row] = [0] * n
for x in range(m):
matrix[x][col] = 0
方法二:标记+数组循环
思路:思路基本上一致,不同在于把原数组转成 array 处理,最后再转回 list 即可。
# 标记+数组循环
import numpy as np
class Solution:
def setZeroes(self, matrix: List[List[int]]) -> None:
m = len(matrix)
n = len(matrix[0])
ans = []
for i in range(m):
for j in range(n):
if matrix[i][j] == 0:
ans.append([i,j])
matrix_arr = np.array(matrix)
for i in ans:
row = i[0]
col = i[1]
matrix_arr[row, :] = [0] * n
matrix_arr[:, col] = [0] * m
matrix[:] = matrix_arr.tolist()
方法三:True 标记法
思路:该算法与前两个算法不同在于,先建立一个 False 列表,元素为 0 位置标记为 True 。再把 True 位置对应的元素更新成 0 即可。
# True 标记法
class Solution:
def setZeroes(self, matrix: List[List[int]]) -> None:
m = len(matrix)
row = [False] * m
n = len(matrix[0])
col = [False] * n
for i in range(m):
for j in range(n):
if matrix[i][j] == 0:
row[i] = col[j] = True
for i in range(m):
for j in range(n):
if row[i] or col[j] is True:
matrix[i][j] = 0
方法四:原地标记
思路:该算法与前面的算法都不同,不需要开辟新的内存空间。只需要锁定第一行和第一列,依次对剩余的行列循环判断即可。最后需要注意的是,对第一行和第一列处理即可。
# 原地标记
class Solution:
def setZeroes(self, matrix: List[List[int]]) -> None:
m = len(matrix)
n = len(matrix[0])
flag_row = any(matrix[i][0] == 0 for i in range(m))
flag_col = any(matrix[0][j] == 0 for j in range(n))
for i in range(1, m):
for j in range(1, n):
if matrix[i][j] == 0:
matrix[i][0] = matrix[0][j] = 0
for i in range(1, m):
for j in range(1, n):
if matrix[i][0] == 0 or matrix[0][j] == 0:
matrix[i][j] = 0
if flag_row:
for i in range(m):
matrix[i][0] = 0
if flag_col:
for j in range(n):
matrix[0][j] = 0
总结
题目有点像小时候玩的炸弹超人啊