This approach uses two additional arrays to track the rows and columns that need to be zeroed.
- Traverse the matrix and mark rows and columns that need to be zeroed.
- Then, update the matrix based on the values in these arrays.
Time Complexity: O(m * n), where m is the number of rows and n is the number of columns.
Space Complexity: O(m + n) for the additional arrays.
1def setZeroes(matrix):
2 m, n = len(matrix), len(matrix[0])
3 row = [False] * m
4 col = [False] * n
5
6 for i in range(m):
7 for j in range(n):
8 if matrix[i][j] == 0:
9 row[i] = True
10 col[j] = True
11
12 for i in range(m):
13 for j in range(n):
14 if row[i] or col[j]:
15 matrix[i][j] = 0This approach uses the matrix itself to track zeroes.
- Use the first row and column as markers.
- Traverse the matrix and if an element is zero, mark its row and column.
- Traverse again and set matrix elements to zero based on marked rows and columns.
Time Complexity: O(m * n), where m is the number of rows and n is the number of columns.
Space Complexity: O(1) since the algorithm runs in constant space.
1public class Solution {
2 public void SetZeroes(int[][] matrix) {
3 int m = matrix.Length;
4 int n = matrix[0].Length;
5 bool isCol = false;
6 for (int i = 0; i < m; i++) {
7 if (matrix[i][0] == 0) isCol = true;
8 for (int j = 1; j < n; j++) {
9 if (matrix[i][j] == 0) {
10 matrix[i][0] = 0;
11 matrix[0][j] = 0;
12 }
13 }
14 }
15
16 for (int i = 1; i < m; i++) {
17 for (int j = 1; j < n; j++) {
18 if (matrix[i][0] == 0 || matrix[0][j] == 0) {
19 matrix[i][j] = 0;
20 }
21 }
22 }
23
24 if (matrix[0][0] == 0) {
25 for (int j = 0; j < n; j++) {
26 matrix[0][j] = 0;
27 }
28 }
29
30 if (isCol) {
31 for (int i = 0; i < m; i++) {
32 matrix[i][0] = 0;
33 }
34 }
35 }
36}