力扣 73. 矩阵置零 思维

https://leetcode-cn.com/problems/set-matrix-zeroes/
在这里插入图片描述
思路:用一个变量记录第一列是否有0,然后就可以把数组的第一行和第一列作为标记位了。如果 m [ i ] [ j ] = 0 m[i][j]=0 m[i][j]=0,就令 m [ i ] [ 0 ] = m [ 0 ] [ j ] = 0 m[i][0]=m[0][j]=0 m[i][0]=m[0][j]=0,标记第 i i i行、第 i i i列需要置为零。最后更新的时候要倒着来或者从第二行开始,因为第一行被用作标记位了,在处理完其它数据之前不能更新,不然可能会有问题,举例:
0 2 0
1 2 3
1 2 3

class Solution {
    
    
public:
    void setZeroes(vector<vector<int>>& matrix) {
    
    
        int n=matrix.size(),m=matrix[0].size();
        bool flag=false;
        for(int i=0;i<n;i++)
        {
    
    
            if(!matrix[i][0])
                flag=true;
            for(int j=1;j<m;j++)
                if(!matrix[i][j])
                    matrix[i][0]=matrix[0][j]=0;
        }
        for(int i=n-1;i>=0;i--)
        {
    
    
            for(int j=1;j<m;j++)
                if(!matrix[i][0]||!matrix[0][j])
                    matrix[i][j]=0;
            if(flag)
                matrix[i][0]=0;
        }
    }
};

猜你喜欢

转载自blog.csdn.net/xiji333/article/details/115038436