lc73-Set Matrix Zeroes

O(1) space解法很妙,把row/col有0的情况记录在第一排/第一列,但(0, 0)的点要注意,不能用来同时表示第一排和第一列,选择一个以后用另外的bool来单独表示另一个。再根据这m+n个0/1来更新全部的格子。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
class Solution {
int m;
public:
void setZeroes(vector<vector<int>>& matrix) {
m = matrix.size();
if(m) {
int n = matrix[0].size();
bool zero;
for (int i = 0; i < m; i++) {
if (!matrix[i][0] ) zero = true;
for (int j = 1; j < n; j++) {
if (!matrix[i][j]) {
cout << i << " " << j << " " << (!i || !j) << endl;
matrix[0][j] = 0;
matrix[i][0] = 0;
}
}
for (int i = m-1; i >= 0; i--) {
for (int j = n-1; j >= 1; j--) {
if (!matrix[i][0] || !matrix[0][j]) {
matrix[i][j] = 0;
}
}
if (zero) matrix[i][0] = 0;
}
}
}
};