Time: 3 ms (71.33%), Space: 18.2 MB (95.28%) - LeetHub

0073-set-matrix-zeroes/0073-set-matrix-zeroes.py

@@ -0,0 +1,21 @@
+class Solution:
+    def setZeroes(self, matrix: List[List[int]]) -> None:
+        """
+        Do not return anything, modify matrix in-place instead.
+        """
+        # O(m+n) space
+        rowsToZero = set()
+        colsToZero = set()
+
+        # Find all zeros
+        for i in range(len(matrix)):
+            for j in range(len(matrix[i])):
+                if matrix[i][j] == 0:
+                    rowsToZero.add(i)
+                    colsToZero.add(j)
+
+        # Zero rows and cols as needed
+        for i in range(len(matrix)):
+            for j in range(len(matrix[i])):
+                if i in rowsToZero or j in colsToZero:
+                    matrix[i][j] = 0