难点是设计一个常数空间的解法。在数组上操作那么就很容易想到利用数组某一行和某一列保存需要归0的信息。这里选的是第一列和第一行。但是第一列不能加入到循环遍历中,不然会被误改成0.所以用了一个标志来确定是不是全变为0. 这样就巧妙地将第一行和第一列作为了信息位而不破坏最终的答案。
class Solution:
def setZeroes(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: void Do not return anything, modify matrix in-place instead.
"""
m=len(matrix)
if m!=0:
n=len(matrix[0])
flag=False
for i in range(m):
if matrix[i][0]==0:
flag=True
for j in range(1,n):
if matrix[i][j]==0:
matrix[i][0],matrix[0][j]=0,0
for i in range(m-1,-1,-1):
for j in range(n-1,0,-1):
if matrix[i][0]==0 or matrix[0][j]==0:
matrix[i][j]=0
if flag:
matrix[i][0]=0