【leetcode】Set Matrix Zeroes

class Solution {
public:
    void setZeroes(vector<vector<int> > &matrix) {
        // Note: The Solution object is instantiated only once and is reused by each test case.
       
        int rowLen=matrix.size();
        int colLen=matrix[0].size();
        
        bool flag1=false;//第一列中有無0
        for(int i=0;i<rowLen;i++)
            if(matrix[i][0]==0)
            {
                flag1=true;
                break;
            }
        
        bool flag2=false;//第一行中有無0
        for(int i=0;i<colLen;i++)
            if(matrix[0][i]==0)
            {
                flag2=true;
                break;
            }
        
        for(int i=1;i<rowLen;i++)//對第二行、第二列及之後的元素進行判斷，當爲0時，反映到對應的第一行、第一列中
            for(int j=1;j<colLen;j++)
            {
                if(matrix[i][j]==0)
                {
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
            }
        
        for(int i=1;i<rowLen;i++)//遍歷除matrix[0][0]外的第一列的元素，來修改其他同一行的元素
            if(matrix[i][0]==0)
                for(int j=1;j<colLen;j++)
                    matrix[i][j]=0;
        
        for(int i=1;i<colLen;i++)//遍歷除matrix[0][0]外的第一行的元素，來修改其他同一列的元素
            if(matrix[0][i]==0)
                for(int j=1;j<rowLen;j++)
                    matrix[j][i]=0;
            
        if(flag1==true)//第一列中有0，則第一行全置爲0
            for(int i=0;i<rowLen;i++)
                matrix[i][0]=0;
        
        if(flag2==true)//第一行中有0，則第一列全置爲0
            for(int i=0;i<colLen;i++)
                matrix[0][i]=0;
    }
};