面试题 01.08. 零矩阵

编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。

输入：
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出：
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]
=================================
输入：
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出：
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/zero-matrix-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

这周学完了集合真的忍不住想用集合写练练手，一开始用的map忘记了key不能重复，又改成set写了一遍

class Solution {
    public void setZeroes(int[][] matrix) {
        Set<int[]> set=new HashSet();
        for(int i=0;i<matrix.length;i++){
            for(int j=0;j<matrix[0].length;j++){
                if(matrix[i][j]==0)set.add(new int[]{i,j});
            }
        }
        Iterator<int[]> iterator = set.iterator();
        while(iterator.hasNext()){
            int[] arr=iterator.next();
            for(int j=0;j<matrix[0].length;j++){
                matrix[arr[0]][j]=0;
            } 
            for(int i=0;i<matrix.length;i++){
                matrix[i][arr[1]]=0; 
            }
        }  
    }
}