Leetcode—73.矩阵置零【中等】

2023每日刷题（六十六）

Leetcode—73.矩阵置零

空间复杂度为O(m+n)版实现代码

class Solution {
   
public:
    void setZeroes(vector<vector<int>>& matrix) {
   
        int rowLen = matrix.size();
        int colLen = matrix[0].size();
        vector<int> row(rowLen, 0);
        vector<int> col(colLen, 0);
        for(int i = 0; i < rowLen; i++) {
   
            for(int j = 0; j < colLen; j++) {
   
                if(matrix[i][j] == 0) {
   
                    row[i] = 1;
                    col[j] = 1;
                }
            }
        }
        for(int i = 0; i < rowLen; i++) {
   
            for(int j = 0; j < colLen; j++) {
   
                if(row[i] || col[j]) {
   
                    matrix[i][j] = 0;
                }
            }
        }
    }
};

运行结果

优化空间复杂度为O(1)版算法思想

优化版实现代码

class Solution {
   
public:
    void setZeroes(vector<vector<int>>& matrix) {
   
        bool rowFlag = false, colFlag = false;
        int rowLen = matrix.size();
        int colLen = matrix[0].size();
        for(int i = 0; i < rowLen; i++) {
   
            if(!matrix[i][0]) {
   
                colFlag = true;
            }
        }

        for(int j = 0; j < colLen; j++) {
   
            if(!matrix[0][j]) {
   
                rowFlag = true;
            }
        }

        for(int i = 1; i < rowLen; i++) {
   
            for(int j = 1; j < colLen; j++) {
   
                if(!matrix[i][j]) {
   
                    matrix[i][0] = matrix[0][j] = 0;
                }
            }
        }

        for(int i = 1; i < rowLen; i++) {
   
            for(int j = 1; j < colLen; j++) {
   
                if(!matrix[i][0] || !matrix[0][j]) {
   
                    matrix[i][j] = 0;
                }
            }
        }

        if(colFlag) {
   
            for(int i = 0; i < rowLen; i++) {
   
                matrix[i][0] = 0;
            }
        }

        if(rowFlag) {
   
            for(int j = 0; j < colLen; j++) {
   
                matrix[0][j] = 0;
            }
        }
    }
};

运行结果

之后我会持续更新，如果喜欢我的文章，请记得一键三连哦，点赞关注收藏，你的每一个赞每一份关注每一次收藏都将是我前进路上的无限动力 ！！！↖(▔▽▔)↗感谢支持！