问题:
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
分析:
这道题的有趣的地方在于,最好能用constant space来解决。一个想法是:用第一行来做indicator,如果某一列里面有0,则对应的第一行的值变为0;相似地,用第一列来做indicator,如果某一行里面有0,则对应的第一列的值变为0。但第一行和第一列本身怎么办呢?我们可以在程序的一开始先来检查第一行和第一列,用两个bool variable来表示。
代码:
class Solution {
public:
void setColZero(vector<vector<int> > &matrix, int col) {
for (int i = 0; i < matrix.size(); i ++)
matrix[i][col] = 0;
}
void setRowZero(vector<vector<int> > &matrix, int row) {
for (int i = 0; i < matrix[0].size(); i ++)
matrix[row][i] = 0;
}
void setZeroes(vector<vector<int> > &matrix) {
bool row = false;
bool col = false;
int m = matrix.size();
if (m == 0)
return;
int n = matrix[0].size();
for (int i = 0; i < n; i ++) {
if (matrix[0][i] == 0) {
row = true;
break;
}
}
for (int i = 0; i < m; i ++) {
if (matrix[i][0] == 0) {
col = true;
break;
}
}
for (int i = 1; i < m; i ++) {
for (int j = 1; j < n; j ++) {
if (matrix[i][j] == 0) {
matrix[0][j] = 0;
matrix[i][0] = 0;
}
}
}
for (int i = 1; i < n; i ++) {
if (matrix[0][i] == 0)
setColZero(matrix, i);
}
for (int i = 1; i < m; i ++) {
if (matrix[i][0] == 0)
setRowZero(matrix, i);
}
if (row)
setRowZero(matrix, 0);
if (col)
setColZero(matrix, 0);
}
};