题目:Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in-place.
这道题乍一看还是蛮简单的,就是将矩阵中,零元素对应的行列所有元素都置为0,如果新开辟一个数组完全可以完成上述要求,但是,难就难在需要in-place完成,说白了空间复杂度只能是O(1),因此,具体的做法只能在原有的位置记录相应的操作,这就带来了一个问题,在遍历矩阵的过程中,当遇到0时,我们无法判断该数是原有矩阵中的0还是我置为后的0,因此,思路是这样的:
对于第一行元素,只需要有一个标志位来标记是否将第一行元素置为0,其余各行元素只要有发现0,就将该元素对应的第一行和第一列的元素都置为0,举个例子,当matrix[i][j] == 0,则:matrix[i][0]和matrix[0][j]都变成0。
第二步,从第2行开始,遍历第1列的元素,发现为0,将该行的所有元素置为0,一定要从第二行开始,否则无法进行后续操作。
第三步,从第1列开始,遍历第一行的所有元素,将第一行中的零元素对应的列全置为0。
第四步,根据设定的标志位来判断第一行是否需要全部置零。
这几步的顺序不能颠倒,尤其是第二步和第三步。
代码如下:
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
bool firstRow = 0;
for (int i=0; i<matrix.size(); ++i) {
for (int j=0; j<matrix[i].size(); ++j) {
if (matrix[i][j] == 0) {
if (!i ){
firstRow = 1;
} else {
matrix[0][j] = 0;
matrix[i][0] = 0;
}
}
}
}
for (int i=1; i<matrix.size(); ++i) {
if (!matrix[i][0]) {
for (int j=1; j<matrix[i].size(); ++j) {
matrix[i][j] = 0;
}
}
}
for (int j=0; j<matrix[0].size(); ++j) {
if (!matrix[0][j]) {
for (int i=1; i<matrix.size(); ++i) {
matrix[i][j] = 0;
}
}
}
if (firstRow) {
for(int i=0; i<matrix[0].size(); ++i) {
matrix[0][i] = 0;
}
}
}
};