思路如下:
1、定义两个HashSet分别表示某行和某列是否存在0;
2、第一次遍历二维数组,若为0,则HashSet中新增对应行数和列数
3、第二次遍历二维数组,若遍历的当前位置的行数或者列数存在于HashSet中,则置0
class Solution {
public void setZeroes(int[][] matrix) {
int m=matrix.length;
int n=matrix[0].length;
HashSet<Integer> row=new HashSet<>();
HashSet<Integer> col=new HashSet<>();
//遍历当前数若为0,则HashSet新增对应行数列数
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(matrix[i][j]==0){
row.add(i);
col.add(j);
}
}
}
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
//若改行或者列已经被标记,则置0
if(row.contains(i)||col.contains(j)){
matrix[i][j]=0;
}
}
}
}
}
原题地址:
73. 矩阵置零