leetcode：73. 矩阵置零

题目来源

• LEETCODE

题目描述

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {

    }
};

题目解析

一边遍历一边修改的难点是前面修改的会影响到后续修改的判断

---

O(m + n)空间复杂度

• 我们不能遇到0后直接对该行该列进行直接修改，因为这样后面出现的0会被“覆盖”，该0所对应的行和列就没有被置0。
• 所以解决办法是将出现0的行和列分别存放在标记数组中，再次遍历矩阵的时将对应出现行和列置为0.
  

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        if(matrix.empty()){
            return;
        }
        int n = matrix.size(), m = matrix[0].size();
        std::vector<int> row(n, 0), col(m, 0);
        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < m; ++j) {
                if(!matrix[i][j]){
                    row[i] = 1;
                    col[j] = 1;
                }
            }
        }

        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < m; ++j) {
                if(row[i] || col[j]){
                    matrix[i][j] = 0;
                }
            }
        }
    }
};

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        if(matrix.empty()){
            return;
        }
        int n = matrix.size(), m = matrix[0].size();
        std::vector<int> row(n), col(m);
        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < m; ++j) {
                if(!matrix[i][j]){
                    row[i] = 1;
                    col[j] = 1;
                }
            }
        }

        for (int i = 0; i < n; ++i) {
            if(row[i] == 1){
                std::fill(matrix[i].begin(), matrix[i].end(), 0);
            }
        }

        for (int i = 0; i < m; ++i) {
            if(col[i] == 1){
                for (int j = 0; j < n; ++j) {
                    matrix[j][i] = 0;
                }
            }
        }
    }
};

O(1)空间复杂度

• 既然标记数组是必不可少的，那么如何才能将空间复杂度降低到O(1)呢？
• 我们只能“拆东墙补西墙”，将原数组的第一行，第一列拆下来作为我们的标记数组。但是这样潜在的问题是，如果第一行或者第一列本来就存在0，那我们需要额外对这一整行进行置0操作
• 注意到第一行既作为标记数组又作为待检查的数组（第一列有额外检查），所以在最终遍历的时候要从后往前。

O(1)空间复杂度

1. 使用两个变量（r0 & c0），记录「首行 & 首列」是否该被置零

2.「非首行首列」的位置

• 将置零信息存储到原矩阵
• 根据置零信息，置零「非首行首列」的位置

3. 使用 r0 & c0 ，置零「首行 & 首列」

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size(), n = matrix[0].size();
        int r0 = 0, c0 = 0;

        // 1. 扫描「首行」和「首列」记录「首行」和「首列」是否该被置零
        for (int i = 0; i < m; ++i) {
            if(matrix[i][0] == 0){
                c0 = 1;
                break;
            }
        }

        for (int j = 0; j < n; ++j) {
            if(matrix[0][j] == 0){
                r0 = 1;
                break;
            }
        }

        // 2.1 扫描「非首行首列」的位置，如果发现零，将需要置零的信息存储到该行的「最左方」和「最上方」的格子内
        for (int i = 1; i < m; ++i) {
            for (int j = 1; j < n; ++j) {
                if(matrix[i][j] == 0)
                    matrix[i][0] = matrix[0][j] = 0;
            }
        }

        // 2.2 根据刚刚记录在「最左方」和「最上方」格子内的置零信息，进行「非首行首列」置零
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                    matrix[i][j] = 0;
                }
            }
        }

        // 3. 根据最开始记录的「首行」和「首列」信息，进行「首行首列」置零
        if(r0 == 1){
            for (int j = 0; j < n; j++) {
                matrix[0][j] = 0;
            }
        }

        if(c0 == 1){
            for (int i = 0; i < m; ++i) {
                matrix[i][0] = 0;
            }
        }
    }
};