LeetCode/矩阵置零

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0

1. O(m+n)辅助空间

基本操作，用两个数组存储对应行和列是否要置零

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

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

2. 常数空间

用第一行和第一列代替方法一的额外空间
其本身信息只需两个常量来记录

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        bool flag1 = false;//表示第一行是否有0
        bool flag2 = false;//表示第一列是否有0
        int rows = matrix.size();
        int cols = matrix[0].size();

        //事先记录第一行状态
        for(int i=0;i<cols;i++)
            if(matrix[0][i]==0)
                flag1 = true;
        //事先记录第一列状态
        for(int i=0;i<rows;i++)
            if(matrix[i][0]==0)
                flag2 = true;
        //以第一行和第一列为标志位，标志是否这一行/列置0
        for(int i=1;i<rows;i++)
            for(int j=1;j<cols;j++)
                if(matrix[i][j]==0)
                    matrix[i][0] = 0,matrix[0][j] = 0;
        
        //把标志位所在列置0
        for(int i=1;i<rows;i++)
            for(int j=1;j<cols;j++)
            if(matrix[i][0]==0|matrix[0][j]==0)
                    matrix[i][j] = 0;
            
        //如果第一行有0存在则将第一行置0
        if(flag1)
            for(int i=0;i<cols;i++)
                matrix[0][i] = 0;
        //如果第一列有0存在则将第一列置0
        if(flag2)
            for(int i=0;i<rows;i++)
                matrix[i][0] = 0;
        return;
    }
}