矩阵置零
问题:
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
思路:
- 第一次遍历数组,用两个标志位数组记录原数组中出现 0 的位置。
flag[i] == 1
表示第 i 行有 0;flag[j] == 0
表示第 j 列有 0;- 再次遍历数组,将有 0 的行和列统统置零。
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
auto m = matrix.size();
auto n = matrix[0].size();
vector<int> flag1(m, 0), flag2(n, 0);
for(auto i = 0; i < m; ++i) {
for(auto j = 0; j < n; ++j) {
if(!matrix[i][j]) {
flag1[i] = flag2[j] = 1;
}
}
}
for(auto i = 0; i < m; ++i) {
for(auto j = 0; j < n; ++j) {
if(flag1[i] || flag2[j]) {
matrix[i][j] = 0;
}
}
}
}
};