方法一:采用两个数组,行数组记录哪些行的所有元素需要全部置为0,列数组需要记录哪些列的所有元素需要全部置为0
public class Solution {public void setZeroes(int[][] matrix) {boolean[] rowArr = new boolean[matrix.length];boolean[] colArr = new boolean[matrix[0].length];for (int i = 0; i < matrix.length; i++) {for (int j = 0; j < matrix[0].length; j++) {if(matrix[i][j]==0){rowArr[i]=true;colArr[j]=true;}}}for (int i = 0; i < rowArr.length; i++)if(rowArr[i])for(int j=0;j<matrix[0].length;j++)matrix[i][j]=0;for(int j=0;j<colArr.length;j++)if(colArr[j])for (int i = 0; i < matrix.length; i++)matrix[i][j]=0;}
}
方法二:在方法一的基础上优化空间复杂度,用第一行和第一列代替方法一中的行数组和列数组,但是这样会导致第一行变动,结果错误,为此引入两个变量,一个记录第一行是否需要全部置为0,另一个记录第一列是否需要全部置为0
public class Solution {public void setZeroes(int[][] matrix) {int m= matrix.length;int n=matrix[0].length;boolean rowZero=false,colZero=false;for (int[] ints : matrix) {if (ints[0] == 0) {colZero = true;break;}}for (int i = 0; i < n; i++) {if(matrix[0][i]==0){rowZero=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++) {for (int j = 1; j < n; j++) {if(matrix[i][0]==0||matrix[0][j]==0)matrix[i][j]=0;}}if(rowZero)for (int i = 0; i < n; i++) {matrix[0][i]=0;}if(colZero)for (int i = 0; i < m; i++) {matrix[i][0]=0;}}
}