LeetCode 73. Set Matrix Zeroes

题目

class Solution {
public:

    void setZeroes(vector<vector<int>>& matrix) {
        
        int tag=0;
        int tag2=0;
       
        for(int i=0;i<matrix.size();i++)
        {
             if(matrix[i][0]==0)
                    tag=1;
            
            for(int j=0;j<matrix[i].size();j++)
            {
                
               if(i==0&&matrix[i][j]==0)
                   tag2=1;
                
                if(matrix[i][j]==0)
                {
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }   
            }
        }
        
        for(int i=1;i<matrix.size();i++)
        {
            for(int j=1;j<matrix[i].size();j++)
            {
                 if(matrix[i][0]==0)
                     matrix[i][j]=0;
                if(matrix[0][j]==0)
                    matrix[i][j]=0;
            }
        }
        
        if(tag2==1)
        {
            for(int j=0;j<matrix[0].size();j++)
                matrix[0][j]=0;
        }
        
        if(tag==1)
        {
            for(int i=0;i<matrix.size();i++)
                matrix[i][0]=0;
        }

    }
};