leetcode Set Matrix Zeroes

Question

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

Solution

If you have to store some thing, but the problem requires in place solution, you may consider storing it in the array given.

class Solution(object):
def setZeroes(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: void Do not return anything, modify matrix in-place instead.
"""
if matrix == None or len(matrix) == 0:
return
first_col = False
first_row = False
for i in xrange(len(matrix[0])):
if matrix[0][i] == 0:
first_row = True
break;
for i in xrange(len(matrix)):
if matrix[i][0] == 0:
first_col = True
break;
for i in xrange(1,len(matrix)):
for j in xrange(1,len(matrix[0])):
if matrix[i][j] == 0:
matrix[i][0] = 0
matrix[0][j] = 0
for i in xrange(1,len(matrix)):
for j in xrange(1,len(matrix[0])):
if matrix[i][0] == 0 or matrix[0][j] == 0:
matrix[i][j] = 0
if first_row:
for i in xrange(len(matrix[0])):
matrix[0][i] = 0
if first_col:
for i in xrange(len(matrix)):
matrix[i][0] = 0