Set Matrix Zeroes

https://leetcode.com/problems/set-matrix-zeroes/

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        self.setZerosOpti(matrix)
        
    def setZerosAdditionalMemory(self, matrix: List[List[int]]) -> None:
        """
        Time Complexity: O(M×N) where M and N are the number of rows 
        and columns respectively

        Space Complexity: O(M+N)
            One array to store zeros in the horizontal axis and another one
            to store the ones from the vertical array.
        """        
        
        horizontal_zeros = []
        vertical_zeros = []
        
        for i, row in enumerate(matrix):
            for j, value in enumerate(row):
                if value == 0:                    
                    horizontal_zeros.append(i)
                    vertical_zeros.append(j)

        
        for i, row in enumerate(matrix):
            for j, value in enumerate(row):
                if i in horizontal_zeros or j in vertical_zeros:
                    matrix[i][j] = 0

        
    def setZerosAdditionalTime(self, matrix: List[List[int]]) -> None:
        """
        Time Complexity: O(M²×N²)
        
        Space Complexity: O(1)
        """
        
        for i, row in enumerate(matrix):
            for j, value in enumerate(row):
                if value == 0:                    
                    print(f"{i}, {j}")
                    # set zeros in row
                    for k, item in enumerate(row):
                        matrix[i][k] = "*" if matrix[i][k] != 0 else 0
                    
                    # set zeros in column
                    for r in matrix:
                        r[j] = "*" if r[j] != 0 else 0
                        
        for i, row in enumerate(matrix):
            for j, value in enumerate(row):
                if value == "*":
                    matrix[i][j] = 0
                    
    def setZerosOpti(self, matrix: List[List[int]]) -> None:    
        """
        Time Complexity: O(M×N)
        
        Space Complexity: O(1)
        """
        
        first_col = False
        num_rows = len(matrix)
        num_cols = len(matrix[0])
        for i in range(num_rows):
            # Since first cell for both first row and first column is the same 
            # i.e. matrix[0][0]
            # We can use an additional variable for either the first row/column.
            # For this solution we are using an additional variable for the first             
            # column and using matrix[0][0] for the first row.
            if matrix[i][0] == 0:
                first_col = True
            for j in range(1, num_cols):
                # If an element is zero, we set the first element of the 
                # corresponding row and column to 0
                if matrix[i][j]  == 0:
                    matrix[0][j] = 0
                    matrix[i][0] = 0
        
        # Iterate over the array once again and using the first row 
        # and first column, update the elements.
        for i in range(num_rows-1 , -1, -1):
            for j in range(num_cols-1, 0, -1):
                if not matrix[i][0] or not matrix[0][j]:
                    matrix[i][j] = 0
            if first_col:
                matrix[i][0] = 0

Last updated