73. Set Matrix Zeroes
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in-place.
Example 1:
Input:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
Output:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
Example 2:
Input:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
Output:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
Follow up:
- A straight forward solution using O(mn) space is probably a bad idea.
- A simple improvement uses O(m + n) space, but still not the best solution.
- Could you devise a constant space solution?
Approach
- 题目大意是某行某列出现0,则将该行该列都设置为零,题目后面说用尽可能少的空间,看到这道题一般很容易想到对行和列做标记,哪行哪列出现零,则将该行和该列进行标记,我第一次做也是这样,先看代码。
Code
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int high = matrix.size(), len = matrix[0].size();
vector<bool>row(high, false);
vector<bool>col(len, false);
for (int i = 0; i < high; i++) {
for (int j = 0; j < len; j++) {
if (matrix[i][j] == 0) {
row[i] = true;
col[j] = true;
}
}
}
for (int i = 0; i < high; i++) {
if (row[i]) {
for (int j = 0; j < len; j++) {
matrix[i][j] = 0;
}
}
else {
for (int j = 0; j < len; j++) {
if (col[j]) {
matrix[i][j] = 0;
}
}
}
}
}
};
Again
- 我没想出怎么用常量的空间,看了别人的代码,原来是用原来的数组腾出一行和一列用作标记。
Code
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int m = matrix.size(), n = matrix[0].size();
bool firstrow = false, firstcol = false;
for (int i = 0; i < m; i++) {
if (matrix[i][0] == 0) {
firstcol = true;
break;
}
}
for (int i = 0; i < n; i++) {
if (matrix[0][i] == 0) {
firstrow = true;
break;
}
}
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (matrix[i][j] == 0) {
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
for (int i = 1; i < m; i++) {
if (matrix[i][0])continue;
for (int j = 1; j < n; j++) {
matrix[i][j] = 0;
}
}
for (int i = 1; i < n; i++) {
if (matrix[0][i])continue;
for (int j = 1; j < m; j++) {
matrix[j][i] = 0;
}
}
if (firstrow) {
for (int i = 0; i < n; i++) {
matrix[0][i] = 0;
}
}
if (firstcol) {
for (int i = 0; i < m; i++) {
matrix[i][0] = 0;
}
}
}
};