题目描述:
https://leetcode-cn.com/problems/set-matrix-zeroes/
题解:
1.一次遍历matrix,记录为0的位置的行序号和列序号,分别保存在zero_row zero_col
2.遍历zero_row,将其中的行全部设为0,遍历zero_col,将其中出现的列全都设为0.
def setZeroes(self, matrix): rowlen = len(matrix) collen = len(matrix[0]) zero_col = [] zero_row = [] for i in range(rowlen): for j in range(collen): if matrix[i][j]==0: if i not in zero_row: zero_row.append(i) if j not in zero_col: zero_col.append(j) for i in range(rowlen): if i in zero_row: for j in range(collen): matrix[i][j] = 0 for i in range(collen): if i in zero_col: for j in range(rowlen): matrix[j][i]=0 return matrix