给定一个 m x n
的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
使用两个数组记录每行,每列有无为零的元素,在遍历一次将有零的行列全置为零,需要遍历两次矩阵
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int m = matrix.size();
int n = matrix[0].size();
vector<int> hang(m); vector<int> lie(n);
for(int i =0; i<m;i++){
for(int j = 0;j<n;j++){
if(matrix[i][j] == 0){
hang[i] = 1;lie[j]=1;
}
}
}
for(int i = 0;i<m;i++){
for(int j=0;j<n;j++){
if(hang[i] || lie[j]){
matrix[i][j] = 0;
}
}
}
}
};