题目描述
题目链接:面试题 01.08. 零矩阵
编写一种算法,若M × N矩阵中某个元素为0,则将其所在的行与列清零。
示例 1
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
示例 2
输入:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
解题核心
找到要处理的行和列,并且要想办法标记,避免同一行/列多次清零的操作
解法一:模拟
找到要清零的行列然后进行操作
public void setZeroes(int[][] matrix) {
int row = matrix.length;//行
int col = matrix[0].length;//列
HashSet<Integer> rowSet = new HashSet<>();
HashSet<Integer> colSet = new HashSet<>();
//记录要清零的行和列(去重减少操作)
for (int i = 0; i < row * col; i++) {
if (matrix[i % row][i / row] == 0) {
rowSet.add(i % row);
colSet.add(i / row);
}
}
//清零
for (int index : rowSet) {
for (int i = 0; i < col; i++) {
matrix[index][i] = 0;
}
}
for (int index : colSet) {
for (int i = 0; i < row; i++) {
matrix[i][index] = 0;
}
}
}