使space o(1)的算法,遍历matrix若为0则matrix[i][0] = matrix[0][j] = 0;
最后从后往前遍历matrix,若该位置matrix[i][0] == 0 || matrix[0][j] == 0 则该位置为0,其中第一列要单独判断,否则会受到matrix[0][0]的干扰
void setZeroes(vector<vector<int>>& matrix) {
int m = matrix.size();
int n=matrix[0].size();
int col0 = 1;
for(int i = 0; i < m; ++ i)
{
if(matrix[i][0] == 0)col0=0;
for(int j = 1; j < n; ++ j)
if(matrix[i][j] == 0)
matrix[i][0] = matrix[0][j] = 0;
}
for(int i = m - 1; i >= 0; -- i)
{
for(int j = 1; j < n; ++ j)
if(matrix[i][0] == 0 || matrix[0][j] == 0)
matrix[i][j] = 0;
if(col0 == 0)
matrix[i][0] = 0;
}
}