题目:
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
示例 1:
输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出:[[1,0,1],[0,0,0],[1,0,1]]
示例 2:
输入:matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出:[[0,0,0,0],[0,4,5,0],[0,3,1,0]]
提示:
m == matrix.length
n == matrix[0].length
1 <= m, n <= 200
-2的31次方 <= matrix[i][j] <= 2的31次方 - 1
进阶:
一个直观的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。
你能想出一个仅使用常量空间的解决方案吗?
解题思路:首先,遍历整个矩阵,当遇到元素值为 0 时,记录下该行和该列需要被置零的情况。创建两个 HashSet
,分别用于记录需要被置零的行和列的索引。第二次遍历矩阵,将记录下来的需要置零的行和列置零。完成后,矩阵中所有包含 0 的行和列都会被置零。
class Solution {
public void setZeroes(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
// 使用两个 Set 分别记录需要被置零的行和列
Set<Integer> rowsToZero = new HashSet<>();
Set<Integer> colsToZero = new HashSet<>();
// 遍历矩阵,记录需要置零的行和列
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (matrix[i][j] == 0) {
rowsToZero.add(i);
colsToZero.add(j);
}
}
}
// 将需要置零的行全部置零
for (int row : rowsToZero) {
for (int j = 0; j < n; j++) {
matrix[row][j] = 0;
}
}
// 将需要置零的列全部置零
for (int col : colsToZero) {
for (int i = 0; i < m; i++) {
matrix[i][col] = 0;
}
}
}
}
仅使用常量空间的解决方案:
使用常量空间意味着要在原始的矩阵中进行标记,以避免使用额外的空间,可以使用矩阵本身的第一行和第一列来标记需要置零的行和列。首先遍历矩阵,当遇到元素值为 0 时,在该元素所在行的第一个位置和所在列的第一个位置做标记,即将对应的第一行和第一列的元素置零。第二次遍历矩阵,根据第一行和第一列的标记情况,将对应行和列的元素置零。最后根据第一行和第一列的标记情况,确定是否需要将第一行和第一列置零。
class Solution {
public void setZeroes(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
boolean firstRowZero = false;
boolean firstColZero = false;
// 标记第一行和第一列是否需要置零
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (matrix[i][j] == 0) {
if (i == 0) firstRowZero = true;
if (j == 0) firstColZero = true;
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 (firstRowZero) {
for (int j = 0; j < n; j++) {
matrix[0][j] = 0;
}
}
if (firstColZero) {
for (int i = 0; i < m; i++) {
matrix[i][0] = 0;
}
}
}
}