LeetCode 73 矩阵置零

矩阵置零

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        //使用回溯法来做
        
        int col=matrix.size();
        if(col==0)
            return;
        int row=matrix[0].size();
        int isRow=false;
        int isCol=false;
        for(int i=0;i<col;++i)
        {
            if(matrix[i][0]==0)
                isCol=true;
        }        
        for(int j=0;j<row;++j)
        {
            if(matrix[0][j]==0)
                isRow=true;            
        }
        for(int i=1;i<col;++i)
        {
            for(int j=1;j<row;++j)
            {
                if(matrix[i][j]==0)
                {
                    matrix[i][0]=0;
                    matrix[0][j]=0;   
                }
            }
            
        }
        for(int i=1;i<col;++i)
        {
            for(int j=1;j<row;++j)
            {
                if(matrix[i][0]==0||matrix[0][j]==0)
                    matrix[i][j]=0;                
            }
        }
        if(isCol==true)
        {
            for(int i=0;i<col;++i)
                matrix[i][0]=0;                     
        }
        if(isRow==true)
            for(int j=0;j<row;++j)
                matrix[0][j]=0;   
    }
};