LeetCode:Set Matrix Zeros

 

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

 

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