162 · 矩阵归零
描述
给定一个m×n矩阵,如果一个元素是0,则将其所在行和列全部元素变成0。
需要在原矩阵上完成操作。
样例
样例 1:
输入:[[1,2],[0,3]]
输出:[[0,2],[0,0]]
样例 2:
输入:[[1,2,3],[4,0,6],[7,8,9]]
输出:[[1,0,3],[0,0,0],[7,0,9]]
挑战
你是否使用了额外的空间?
一个直接的解决方案是使用 O(MN) 的额外空间,但这并不是一个好的方案。
一个简单的改进方案是使用 O(M + N) 的额外空间,但这仍然不是最好的解决方案。
你能想出一个常数空间的解决方案吗?
void setZeroes(vector<vector<int>> &matrix) {
// write your code here
int rowSize = matrix.size();
if(rowSize <=0)
return ;
int colSize = matrix[0].size();
vector<bool> rowVec(rowSize,false);
vector<bool> colVec(colSize,false);
for(int i =0;i<rowSize;i++)
{
for(int j=0;j<colSize;j++)
{
if(matrix[i][j] == 0 )
{
rowVec[i] = true;
colVec[j] = true;
}
}
}
for(int i =0;i<rowSize;i++)
{
for(int j=0;j<colSize;j++)
{
if(true == rowVec[i] || true == colVec[j])
{
matrix[i][j] = 0;
}
}
}
}