题号:no8
题目名:零矩阵
题目描述
编写一种算法,若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]
]
思路
1.遍历矩阵,将为0的行和列存入到set中
2.遍历set,将set里面有的行和列全部变成0
解题代码
public class Solution {
public void setZeroes(int[][] matrix) {
//遍历矩阵,如果该位置为0,将该位置所在行列存入set集合
if(matrix==null||matrix.length==0) return;
Set<Integer> rows = new HashSet<>();
Set<Integer> cols = new HashSet<>();
for (int i = 0; i < matrix.length; i++) {
for (int j = 0; j < matrix[0].length; j++) {
if(matrix[i][j]==0) {
rows.add(i);
cols.add(j);
}
}
}
//重新遍历矩阵,修改矩阵所在行和列
for (Integer row : rows) {
Arrays.fill(matrix[row],0);
}
for (Integer col : cols) {
for (int i = 0; i < matrix.length; i++) {
matrix[i][col]=0;
}
}
}
}