Zero Matrix
Make INPLACE row and column of a matrix 0 iff that row contains a 0
def zero_matrix(matrix):
"""
Make INPLACE row and column of a matrix 0 iff that row contains a 0
"""
position_of_zeros = []
row = len(matrix)
col = len(matrix[0])
for i in range(0, row):
for j in range(0, col):
if matrix[i][j] == 0:
position_of_zeros.append((i, j))
for (i, j) in position_of_zeros:
for k in range(0, col):
matrix[i][k] = 0
for k in range(0, row):
matrix[k][j] = 0
return matrix
if __name__ == "__main__":
matrix_1 = zero_matrix([[0, 2, 3, 4], [5, 6, 7, 8], [9, 10, 0, 12], [13, 14, 15, 16]])
assert matrix_1 == [[0, 0, 0, 0], [0, 6, 0, 8], [0, 0, 0, 0], [0, 14, 0, 16]]
matrix_2 = zero_matrix([[0, 2, 3, 4], [5, 6, 7, 8], [9, 10, 0, 12], [13, 14, 15, 16], [1, 2, 3, 0]])
assert matrix_2 == [[0, 0, 0, 0], [0, 6, 0, 0], [0, 0, 0, 0], [0, 14, 0, 0], [0, 0, 0, 0]]
Updated on 2020-03-19