May 21

Author: siddydutta

2025-05-May-LeetCoding-Challenge/README.md

@@ -24,7 +24,7 @@
 | May 18 | [1931. Painting a Grid With Three Different Colors](https://leetcode.com/problems/painting-a-grid-with-three-different-colors/) | Hard | Unsolved |
 | May 19 | [3024. Type of Triangle](https://leetcode.com/problems/type-of-triangle/) | Easy | Solved |
 | May 20 | [3355. Zero Array Transformation I](https://leetcode.com/problems/zero-array-transformation-i/) | Medium | Solved |
-| May 21 | []() |  |  |
+| May 21 | [73. Set Matrix Zeroes](https://leetcode.com/problems/set-matrix-zeroes/) | Medium | Solved |
 | May 22 | []() |  |  |
 | May 23 | []() |  |  |
 | May 24 | []() |  |  |
@@ -41,5 +41,5 @@
 | Level  | Problems | Solved | Unsolved |
 | ---    | --- | --- | --- |
 | Easy   | 5 | 5 | 0 |
-| Medium | 10 | 8 | 2 |
+| Medium | 11 | 9 | 2 |
 | Hard   | 4 | 1 | 3 |

2025-05-May-LeetCoding-Challenge/Set Matrix Zeroes.py

@@ -0,0 +1,60 @@
+from itertools import product
+
+
+class Solution:
+    def setZeroes(self, matrix: list[list[int]]) -> list[list[int]]:
+        rows, cols = len(matrix), len(matrix[0])
+        first_row_zero, first_col_zero = False, False
+
+        for row, col in product(range(rows), range(cols)):
+            if matrix[row][col] == 0:
+                if row == 0:
+                    first_row_zero = True
+                if col == 0:
+                    first_col_zero = True
+                matrix[row][0] = None
+                matrix[0][col] = None
+
+        for row in range(1, rows):
+            if matrix[row][0] is None:
+                for col in range(cols):
+                    matrix[row][col] = 0
+        for col in range(1, cols):
+            if matrix[0][col] is None:
+                for row in range(rows):
+                    matrix[row][col] = 0
+
+        if first_col_zero is True:
+            for row in range(rows):
+                matrix[row][0] = 0
+        if first_row_zero is True:
+            for col in range(cols):
+                matrix[0][col] = 0
+
+
+def main():
+    matrix = [[1, 1, 1], [1, 0, 1], [1, 1, 1]]
+    Solution().setZeroes(matrix)
+    assert matrix == [[1, 0, 1], [0, 0, 0], [1, 0, 1]]
+
+    matrix = [[0, 1, 2, 0],
+              [3, 4, 5, 2],
+              [1, 3, 1, 5]]
+    Solution().setZeroes(matrix)
+    assert matrix == [[0, 0, 0, 0],
+                      [0, 4, 5, 0],
+                      [0, 3, 1, 0]]
+
+    matrix = [[1, 2, 3, 4],
+              [5, 0, 7, 8],
+              [0, 10, 11, 12],
+              [13, 14, 15, 0]]
+    Solution().setZeroes(matrix)
+    assert matrix == [[0, 0, 3, 0],
+                      [0, 0, 0, 0],
+                      [0, 0, 0, 0],
+                      [0, 0, 0, 0]]
+
+
+if __name__ == '__main__':
+    main()