浅谈 矩阵置零 问题

矩阵置零

问题：
给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。

思路：

1. 第一次遍历数组，用两个标志位数组记录原数组中出现 0 的位置。
2. flag[i] == 1 表示第 i 行有 0；
3. flag[j] == 0 表示第 j 列有 0；
4. 再次遍历数组，将有 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;
                }
            }
        }
    }
};