首先我们需要两个数组(这里使用set可以剔除重复元素),分别标记包含零的行和列,然后将这些行和列全部置零就行了。
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
// 用于包含零的行与列
set<int> row;
set<int> col;
for(int i=0;i<matrix.size();i++){
for(int j=0;j<matrix[i].size();j++){
if(matrix[i][j] == 0){
row.insert(i);
col.insert(j);
}
}
}
for(auto it = row.begin();it!=row.end();it++){
// 将包含零的行全部置为零
for(int j=0;j<matrix[*it].size();j++){
matrix[*it][j] = 0;
}
}
for(auto it = col.begin();it!=col.end();it++){
// 将包含零的列全部置为零
for(int i=0;i<matrix.size();i++){
matrix[i][*it] = 0;
}
}
}
};