束手无策。题意看起来是简单的,但是实行起来是困难的。matrix 是行的集合,换句话说,就是一个二维数组里面存了行,很多个行,matrix 0 存的是第 0 行。其实有点难。但是我一定可以的。我是可以的。我一遍一遍地告诉自己,我一定可以,直到有一天自己真的做到了,或者彻底放弃了。但是我不可能放弃的。我一定可以做到的。啥意思,就是标记,标记可能的情况,如果遍历到这个元素是 0 那么就标记这个元素所在的行和列,全部标记到第一行和第一列,然后遍历整个二维数组,最后特殊处理第一列和第一行。
cpp
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int m = matrix.size();
int n = matrix[0].size();
bool firstCow = false;
bool firstCol = false;
for ( int i = 0; i < n; i++ ) {
if ( matrix[0][i] == 0 ) {
firstCow = true;
break;
}
}
for ( int i = 0; i < m; i++ ) {
if ( matrix[i][0] == 0 ) {
firstCol = 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++ ) {
for ( int j = 1; j < n; j++ ) {
if ( matrix[i][0] == 0 || matrix[0][j] == 0 ) {
matrix[i][j] = 0;
}
}
}
if ( firstCol ) {
for ( int i = 0; i < m; i++ ) {
matrix[i][0] = 0;
}
}
if ( firstCow ) {
for ( int i = 0; i < n; i++ ) {
matrix[0][i] = 0;
}
}
}
};