LeetCode 73. 矩阵置零

题目链接

力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

题目解析

1、定义两个变量，flag_col0与flag_row0。分别代表第一列和第一行是否为0。我们只需要遍历第一行和第一列，只要遇见0则对应的第一行和第一列就赋值为true.

2、我们使用第一行和第一列的数来去记录该行是否出现0，和该列是否出现0.若某行出现0，则该数所对应的第一行与第一列的数设为0.

3、根据第一行与第一列的情况来对整个数组(除了第一行和第一列)进行赋值。

4、根据flag_col0与flag_row0来去对第一行和第一列进行赋值。

代码

class Solution 
{
public:
    void setZeroes(vector<vector<int>>& matrix)
    {
        int m=matrix.size();
        int n=matrix[0].size();
        // flag_col0 记录第一列是否出现0 也就是第一列是否需要全部赋值为0(只要有一个为0最终都为0)
        // flag_row0 记录第一行是否出现0 也就是第一行是否需要全部赋值为0(只要有一个为0最终都为0)
        // 第一列用来记录每行是否出现0 
        // 第一行用来记录每列是否出现0
        // 最终使用flag_col0与flag_row0重新对第一列的数据和第一行的数据进行赋值

        int flag_col0 = false, flag_row0 = false;
        // 判断flag_col0
        for(int i=0;i<m;i++)
        {
            if(!matrix[i][0]) flag_col0=true;
        }
        // 判断flag_row0
        for(int j=0;j<n;j++)
        {
            if(!matrix[0][j]) flag_row0=true;
        }
        // 遍历每个元素，若该元素为0，matrix[i][0]=matrix[0][j]=0
        for(int i = 1; i < m; i++)
            for(int j = 1; j<n;j++)
            {
                if(!matrix[i][j])
                {
                    matrix[i][0]=matrix[0][j]=0;
                }
            }
        // 根据我们得到的第一行与第一列是否为0，来对数据重新赋值
        for(int i = 1; i < m; i++)
            for(int j = 1; j<n;j++)
            {
                if(!matrix[i][0]||!matrix[0][j])
                {
                    matrix[i][j]=0;
                }
            }
        // 对第一行和第一列重新赋值
        if(flag_col0)
        {
            for(int i=0;i<m;i++)
                matrix[i][0]=0;
        }
        if(flag_row0)
        {
            for(int i=0;i<n;i++)
             matrix[0][i]=0;
        }
       
    }
};