题目链接

力扣（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_col0for(int i=0;i<m;i++){if(!matrix[i][0]) flag_col0=true;}// 判断flag_row0for(int j=0;j<n;j++){if(!matrix[0][j]) flag_row0=true;}// 遍历每个元素，若该元素为0，matrix[i][0]=matrix[0][j]=0for(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;}}
};