解题思路:
本题要求原地置换元素
对矩阵进行第一轮遍历,使用第一行第一列来充当该行该列是否要置换为0的标记位,如果第一行或第一列本身就含有零元素,我们使用colZero和rowZero变量来对其标记。如果第i行第j列的那个元素为0,我们就将[i][0]和[0][j]的元素置换为0,接着判断i和j是否为0,即第一行第一列,如果是,则colZero和rowZero都赋值为true。
对矩阵进行第二轮遍历:即对不包含第一行或第一列进行处理,根据第一轮遍历得到的标记将对应的行和列中的元素置换为0。
对矩阵进行第三轮遍历:即对第一行和第一列本身就含有0元素的情况进行处理,如果第一行或者第一列本身就包含0元素,则将其对应的行或列元素都置换为0。
代码实现:
public void setZeroes(int[][] matrix) {int m=matrix.length;int n=matrix[0].length;boolean colZero=false;//第一行是否有0boolean rowZero=false;//第一列是否有0for(int i=0;i<m;i++){for(int j=0;j<n;j++){if(matrix[i][j]==0){matrix[0][j]=matrix[i][0]=0;//标记位//判断是否第一行第一列为0if(i==0) rowZero=true;if(j==0) colZero=true;}}}//第二轮遍历,将标记的行和列置换为0for(int i=1;i<m;i++){for(int j=1;j<n;j++){if(matrix[i][0]==0 || matrix[0][j]==0){matrix[i][j]=0;}}}//如果第一行或第一列本来就有0,则进行0置换for(int i=0; colZero && i<m;i++) matrix[i][0]=0;for(int j=0; rowZero && j<n;j++) matrix[0][j]=0;}