73. Set Matrix Zeroes

题目:Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in-place.

这道题乍一看还是蛮简单的,就是将矩阵中,零元素对应的行列所有元素都置为0,如果新开辟一个数组完全可以完成上述要求,但是,难就难在需要in-place完成,说白了空间复杂度只能是O(1),因此,具体的做法只能在原有的位置记录相应的操作,这就带来了一个问题,在遍历矩阵的过程中,当遇到0时,我们无法判断该数是原有矩阵中的0还是我置为后的0,因此,思路是这样的:

对于第一行元素,只需要有一个标志位来标记是否将第一行元素置为0,其余各行元素只要有发现0,就将该元素对应的第一行和第一列的元素都置为0,举个例子,当matrix[i][j] == 0,则:matrix[i][0]和matrix[0][j]都变成0。

第二步,从第2行开始,遍历第1列的元素,发现为0,将该行的所有元素置为0,一定要从第二行开始,否则无法进行后续操作。

第三步,从第1列开始,遍历第一行的所有元素,将第一行中的零元素对应的列全置为0。

第四步,根据设定的标志位来判断第一行是否需要全部置零。

这几步的顺序不能颠倒,尤其是第二步和第三步。

代码如下:

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        bool firstRow = 0;
        
        for (int i=0; i<matrix.size(); ++i) {
            for (int j=0; j<matrix[i].size(); ++j) {
                if (matrix[i][j] == 0) {
                    if (!i ){
                        firstRow = 1;
                    } else {
                        matrix[0][j] = 0;
                        matrix[i][0] = 0;   
                    }
                }
            }
        }
        
        for (int i=1; i<matrix.size(); ++i) {
            if (!matrix[i][0]) {
                for (int j=1; j<matrix[i].size(); ++j) {
                    matrix[i][j] = 0;
                }
            }
        }
        
        for (int j=0; j<matrix[0].size(); ++j) {
            if (!matrix[0][j]) {
                for (int i=1; i<matrix.size(); ++i) {
                    matrix[i][j] = 0;
                }
            }
        }
        
        if (firstRow) {
            for(int i=0; i<matrix[0].size(); ++i) {
                matrix[0][i] = 0;
            }
        }
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值