面试题 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]
]

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;
            }
        }
        
    }
}