两种做法:
1:m+n的空间复杂度,定义两个bool数组,分别为row和col,记录这个row和col是否要置0,最后再遍历这两个数组,将matrix的相应值置0;
2:1的空间复杂度,利用第一行和第一列来记录从第二行和第二列开始的矩阵中的某一位置是否置为0.
public class Solution {
public void setZeroes(int[][] matrix) {
if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
return;
}
boolean empty_row = false;
boolean empty_col = false;
int r = matrix.length, c = matrix[0].length;
for (int i = 0; i < r; i++) {
if (matrix[i][0] == 0) {
empty_row = true;
break;
}
}
for (int i = 0; i < c; i++) {
if (matrix[0][i] == 0) {
empty_col = true;
break;
}
}
for (int i = 1; i < r; i++) {
for (int j = 1; j < c; j++) {
if (matrix[i][j] == 0) {
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
for (int i = 1; i < r; i++) {
for (int j = 1; j < c; j++) {
if (matrix[i][0] == 0 || matrix[0][j] == 0) {
matrix[i][j] = 0;
}
}
}
if (empty_row) {
for (int i = 0; i < r; i++) {
matrix[i][0] = 0;
}
}
if (empty_col) {
for (int i = 0; i < c; i++) {
matrix[0][i] = 0;
}
}
}
// public void setZeroes(int[][] matrix) {
// if (matrix == null || matrix.length == 0 || matrix[0].length == 0) {
// return;
// }
// boolean []row = new boolean[matrix.length];
// boolean []col = new boolean[matrix[0].length];
// for (int i = 0; i < row.length; i++) {
// for (int j = 0; j < col.length; j++) {
// if (matrix[i][j] == 0) {
// row[i] = true;
// col[j] = true;
// }
// }
// }
// for (int i = 0; i < row.length; i++) {
// for (int j = 0; j < col.length; j++) {
// if (row[i] || col[j]) {
// matrix[i][j] = 0;
// }
// }
// }
// }
}