https://leetcode.cn/problems/set-matrix-zeroes/description/?envType=study-plan-v2&envId=top-100-liked
我们可以使用两个vector来记录需要置零的行和列,然后遍历这两个vector,将对应的行和列置零。这样的空间复杂度最大是O(n+m)
时间复杂度O(nm),空间复杂度O(n+m)
java
public void setZeroes(int[][] matrix) {
HashSet<Integer> row = new HashSet<>();
HashSet<Integer> col = new HashSet<>();
for (int i = 0; i < matrix.length; i++) {
for (int j = 0; j < matrix[0].length; j++) {
if (matrix[i][j] == 0) {
row.add(i);
col.add(j);
}
}
}
for(int i = 0; i < matrix.length; i++) {
for(int j = 0; j < matrix[0].length; j++) {
if (row.contains(i) || col.contains(j)) {
matrix[i][j] = 0;
}
}
}
}
因为当我们记录i,j是需要置0的行和列时,我们可以使用原矩阵的首行和首列置0来代替集合记录,因为后续操作这些位置也要置0的.时间复杂度O(nm),空间复杂度O(1)
java
public void setZeroes(int[][] matrix) {
// 定义两个布尔变量,分别表示第一列和第一行是否有0
boolean col0 = false, row0 = false;
// 遍历第一列,如果有0,则将col0设为true
for (int i = 0; i < matrix.length; i++) {
if (matrix[i][0] == 0) col0 = true;
}
// 遍历第一行,如果有0,则将row0设为true
for (int i = 0; i < matrix[0].length; i++) {
if (matrix[0][i] == 0) row0 = true;
}
// 遍历除第一行和第一列外的所有元素,如果有0,则将对应的第一行和第一列元素设为0
for (int i = 1; i < matrix.length; i++) {
for (int j = 1; j < matrix[0].length; j++) {
if (matrix[i][j] == 0) {
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
// 遍历第一列,如果有0,则将对应的所有行元素设为0
for (int i = 1; i < matrix[0].length; i++) {
if (matrix[0][i] == 0) {
for (int j = 1; j < matrix.length; j++) {
matrix[j][i] = 0;
}
}
}
// 遍历第一行,如果有0,则将对应的所有列元素设为0
for (int i = 1; i < matrix.length; i++) {
if (matrix[i][0] == 0) {
for (int j = 1; j < matrix[0].length; j++) {
matrix[i][j] = 0;
}
}
}
// 如果第一行有0,则将第一行所有元素设为0
if(row0) {
Arrays.fill(matrix[0], 0);
}
// 如果第一列有0,则将第一列所有元素设为0
if(col0) {
for(int i = 0; i < matrix.length; i++) {
matrix[i][0] = 0;
}
}
}