算法思想
肯定要另外开辟空间的,要不然遍历一遍,将行和列置0,那么后面就不知道到底是原来的0还是后面置的0
如果使用O(mn)的空间,那么需要存放原来0的位置
如果使用O(m + n)的空间,那么需要更新每一行和每一列0的位置
有一种思想就是先记录第一行和第一列有没有0
然后从(1,1)开始遍历,如果遍历到0的话,在第一行和第一列更新,进行记录,就不需要开辟O(m+n)的空间了。
最后在根据第一行和第一列有没有0来更新行和列,这样空间复杂度为O(1)
class Solution:
def setZeroes(self, matrix: List[List[int]]) -> None:
"""
Do not return anything, modify matrix in-place instead.
"""
rowFlage, colFlage = False, False
rows, cols = len(matrix), len(matrix[0])
for i in range(rows):
if matrix[i][0] == 0:
rowFlage = True
break
for j in range(cols):
if matrix[0][j] == 0:
colFlage = True
break
for i in range(rows):
for j in range(cols):
if matrix[i][j] == 0:
matrix[i][0] = 0
matrix[0][j] = 0
for i in range(1, rows):
if matrix[i][0] == 0:
for j in range(1, cols):
matrix[i][j] = 0
for j in range(1, cols):
if matrix[0][j] == 0:
for i in range(1, rows):
matrix[i][j] = 0
if rowFlage:
for i in range(rows):
matrix[i][0] = 0
if colFlage:
for j in range(cols):
matrix[0][j] = 0