LeetCode: Set Matrix Zeroes [073]

class Solution {
public:
    void setZeroes(vector<vector<int> > &matrix) {
        int rows=matrix.size();
        if(rows==0)return;
        int cols=matrix[0].size();
        if(cols==0)return;
        
        bool isR1Removed=false;
        bool isC1Removed=false;
        // 判断第1行是否需要清零
        for(int j=0; j<cols; j++){
            if(matrix[0][j]==0){isR1Removed=true;break;}
        }
        // 判断第1列是否需要清零
        for(int i=0; i<rows; i++){
            if(matrix[i][0]==0){isC1Removed=true;break;}
        }
        
        //扫描除第一行第一列之外的区域，并在第一行和第一列做相应的清零标记
        for(int i=1; i<rows; i++){
            for(int j=1; j<cols; j++){
                if(matrix[i][j]==0){
                    matrix[0][j]=0;
                    matrix[i][0]=0;
                }
            }
        }
        
        //扫描第一行，相应列清零
        for(int j=1; j<cols; j++){
            if(matrix[0][j]==0){
                for(int i=1; i<rows; i++){
                    matrix[i][j]=0;
                }
            }
        }
        //扫描第一列，相应行清零
        for(int i=1; i<rows; i++){
            if(matrix[i][0]==0){
                for(int j=1; j<cols; j++){
                    matrix[i][j]=0;
                }
            }
        }
        if(isR1Removed){
            //第一行清0
            for(int j=0; j<cols; j++) matrix[0][j]=0;
        }
        if(isC1Removed){
            //第一列清0
            for(int i=0; i<rows; i++) matrix[i][0]=0;
        }

    }
};