思路:先遍历一遍数组,用布尔数组记录下有0元素的行和列,然后再遍历数组,利用布尔数组对对应元素置0。
题目要求用O(m+n)的空间复杂度,所以不能用二维布尔数组,应用两个一维布尔数组,一个记录行出现0的位置,一个记录列出现0的位置。
public void setZeroes(int[][] matrix) {
int rowsize = matrix.length;
int columnsize = 0;
if(rowsize > 0)columnsize = matrix[0].length;
boolean row[] = new boolean[rowsize];
boolean column[] = new boolean[columnsize];
for(int i = 0;i<rowsize;i++){
for(int j = 0;j<columnsize;j++){
if(matrix[i][j] == 0){
row[i] = true;
column[j] = true;
}
}
}
for(int i = 0;i<rowsize;i++){
for(int j = 0;j<columnsize;j++){
if(row[i] || column[j])matrix[i][j]=0;
}
}
}