面试题 01.08. 零矩阵 #
题目 #
- 编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。
思路 #
模拟 #
代码 #
模拟 #
class Solution {
public void setZeroes(int[][] matrix) {
Set<Integer> row = new HashSet<>(), 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 r: row) {
for (int j = 0; j < matrix[0].length; j++) {
matrix[r][j] = 0;
}
}
for (int c: col) {
for (int i = 0; i < matrix.length; i++) {
matrix[i][c] = 0;
}
}
}
}