这题本来想的挺好,觉得只要两个数组,一个记录哪行出现过0,一个记录哪列出现个0就好了,结果发现原来不能额外空间...
所以用第一行和第一列代替了,先查看第一列与第一行是否有0元素,之后可按之前的行列数组法对剩余数组进行处理。
代码如下:
class Solution {
public:
void setZeroes(vector<vector<int> > &matrix) {
int m = matrix.size();
int n = matrix[0].size();
int i=0,j=0,row=0,col=0;
for(;i<m;++i){
if(matrix[i][0]==0){
col = 1;
break;
}
}
for(;j<n;++j){
if(matrix[0][j]==0){
row = 1;
break;
}
}
for(i=1;i<m;++i){
for(j=1;j<n;++j){
if(matrix[i][j]==0){
matrix[0][j]=0;
matrix[i][0]=0;
}
}
}
for(i=1;i<m;++i){
for(j=1;j<n;++j){
if(matrix[i][0]==0||matrix[0][j]==0){
matrix[i][j]=0;
}
}
}
if(col){
for(i=0;i<m;++i){
matrix[i][0]=0;
}
}
if(row){
for(j=0;j<n;++j){
matrix[0][j]=0;
}
}
}
};