73.Set Matrix Zeroes

思路:
  • 设置两个bool数组row[matrix[0].size()],col[matrix.size()]分别记录行和列中的0,如果matrix[i][j]为0,那么将row[j]和col[i]赋值为0。先遍历数组matrix,更新bool数组,然后遍历bool数组,更新matrix。(也可以将0标记到第一行和第一列,这样空间复杂度为(O(1)),以后更新)
class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        vector<bool>row(matrix[0].size(),false);
        vector<bool>col(matrix.size(),false);
        for(int i = 0; i < matrix.size(); i++){
            for(int j = 0; j < matrix[0].size(); j++){
                if(matrix[i][j] == 0){
                    row[j] = true;
                    col[i] = true;
                }
            }
        }
        for(int i = 0; i < matrix.size(); i++){
            if(col[i] == true){
                for(int j = 0; j < matrix[0].size(); j++){
                    matrix[i][j] = 0;
                }
            }
        }
        for(int i = 0; i < matrix[0].size(); i++){
            if(row[i] == true){
                for(int j = 0; j < matrix.size(); j++){
                    matrix[j][i] = 0;
                }
            }
        }
        
    }
};
原文地址:https://www.cnblogs.com/UniMilky/p/6985989.html