Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
要求空间复杂度为o(1)
思路:把数组划分为两部分,第一部分是第一行和第一列;第二部分是除去第一行第一列的矩阵部分。
1.先记录第一行第一列是否需要清零。
2.用第一行来存储第二部分矩阵需要清零的列,第一列来存储第二部分矩阵需要清零的行,例如,如果matrix[i][j] = 0,那么令matrix[i][0]=0,matrix[0][j] = 0,其中i>0,j>0,表示i行需要清零,第j列需要清零。
3.标记需要清0的行列
4.对行列清0
5.根据1,对第一行第一列清零
class Solution { public: void setZeroes(vector<vector<int>>& matrix) { int m = matrix.size(); if(m==0){ return ; } int n = matrix[0].size(); bool isFirstRowHasZero = false; bool isFirstColHasZero = false; for(int j=0;j<n;j++){ if(matrix[0][j]==0){ isFirstRowHasZero = true; break; } } for(int i=0;i<m;i++){ if(matrix[i][0]==0){ isFirstColHasZero = true; break; } } for(int i=1;i<m;i++){ for(int j=1;j<n;j++){ if(matrix[i][j]==0){ matrix[i][0] = 0; matrix[0][j] = 0; } } } for(int i=1;i<m;i++){ if(matrix[i][0] == 0){ for(int j=1;j<n;j++){ matrix[i][j] = 0; } } } for(int j=1;j<n;j++){ if(matrix[0][j] == 0){ for(int i=1;i<m;i++){ matrix[i][j] = 0; } } } if(isFirstRowHasZero){ for(int j=0;j<n;j++){ matrix[0][j] = 0; } } if(isFirstColHasZero){ for(int i=0;i<m;i++){ matrix[i][0] = 0; } } } };