面试题 01.08. 零矩阵
编写一种算法,若M × N矩阵中某个元素为0,则将其所在的行与列清零。
示例 1:
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
思路
增加一个辅助空间
class Solution {
private:
void change(vector<vector<int>>& matrix, int row, int col){
for(int i=0;i<matrix.size();i++) matrix[i][col]=0;
for(int i=0;i<matrix[row].size();i++) matrix[row][i]=0;
}
public:
void setZeroes(vector<vector<int>>& matrix) {
vector<vector<int> > copy = matrix;
for(int i=0;i<copy.size();i++){
for(int j=0;j<copy[i].size();j++){
if(copy[i][j]==0)
change(matrix, i, j);
}
}
}
};