题目链接
思路1
遍历矩阵,分别使用集合 row
和 column
记录值为0的行和列。最后将 row
和 column
所记录的行和列置为零。
空间复杂度: \(O(m + n)\)
代码1
class Solution {
public void setZeroes(int[][] matrix) {
Set<Integer> row = new HashSet<>();
Set<Integer> column = new HashSet<>();
for(int i = 0; i < matrix.length; i++){
for(int j = 0; j < matrix[i].length; j++){
if(matrix[i][j] == 0){
row.add(i);
column.add(j);
}
}
}
for(int i : row){
Arrays.fill(matrix[i], 0);
}
for(int j : column){
for(int i = 0; i < matrix.length; i++){
matrix[i][j] = 0;
}
}
}
}