编写一种算法,若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]
]
class Solution {
public void setZeroes(int[][] matrix) {
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(cols.contains(j)) continue;
if(matrix[i][j]==0){
rows.add(i);
cols.add(j);
}
}
}
for(int row:rows){
for(int j=0; j<matrix[0].length; j++){
matrix[row][j]=0;
}
}
for(int col:cols){
for(int j=0; j<matrix.length; j++){
matrix[j][col]=0;
}
}
}
}