力扣73. 矩阵置零

染色

• 思路：
  • 先遍历一遍数组，如果元素为 0 ，将当前行和列染色标记，使用两个数组存储染色标记位；
  • 再遍历一遍数组，如果当前行或者列被染色，将当前元素值置零；

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int row = matrix.size();
        if (row == 0) {
            return;
        }
        int column = matrix[0].size();
        if (0 == column) {
            return;
        }

        std::vector<int> rflag(row);
        std::vector<int> cflag(column);
        for (int i = 0; i < row; ++i) {
            for (int j = 0; j < column; ++j) {
                if (matrix[i][j] == 0) {
                    rflag[i] = 1;
                    cflag[j] = 1;
                }
            }
        }

        for (int i = 0; i < row; ++i) {
            for (int j = 0; j < column; ++j) {
                if (rflag[i] || cflag[j]) {
                    matrix[i][j] = 0;
                }
            }
        }
    }
};

• 该算法空间复杂度为 O(m + n)，一行和一列来存储标记位；有空间复杂度 O(1) 的算法，复用原有矩阵空间，后续可以深入研究一下；