矩阵对应的行列设置为零

思路：将对应的对应的0都落到第一行和第一列上，不过之前要判断一下第一行和第一列是否有0。

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

 

使用辅助空间

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