leetcode 73. Set Matrix Zeroes

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in-place.

Example 1:

Input:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
Output:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
让把0所在的行和列全部都设为0，而且要在matrix自身上变动元素

思路：
可以看到如果遇到0元素而直接设行和列为0的话，后面0的数量会大大增多而影响time complexity
所以较好的办法是先把0所在的行和列保存起来，然后再一次性设置matrix

第一种方法，用两个hashSet分别保存0元素所在的行和列
因为hashSet是根据hashMap实现的，插入元素时要计算位置，而且后面按行和列更新元素，会有重复访问的元素，所以耗时2ms，40%

//2ms	
    public void setZeroes(int[][] matrix) {
        if (matrix == null || matrix.length == 0) {
            return;
        }
        
        HashSet<Integer> hRow = new HashSet<>();
        HashSet<Integer> hCol = new HashSet<>();
        
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[0].length; j++) {
               if (matrix[i][j] == 0) {
                   hRow.add(i);
                   hCol.add(j);
               } 
            }
        }
        
        for (Integer i : hRow) {
            for (int j = 0; j < matrix[0].length; j++) {
                matrix[i][j] = 0;
            }
        }
        
        for (Integer j : hCol) {
            for (int i = 0; i < matrix.length; i++) {
                matrix[i][j] = 0;
            }
        }
        
        return;
    }

为了进一步削减时间复杂度，参考了大神的方法，采用数组保存0所在行和列的flag，O(1)时间访问，而且同时看行和列的flag，避免了重复访问
时间改进到1ms，100%

    public void setZeroes(int[][] matrix) {
        if (matrix == null || matrix.length == 0) {
            return;
        }
        
        int m = matrix.length;
        int n = matrix[0].length;
        
        boolean[] bRow = new boolean[m];
        boolean[] bCol = new boolean[n];
        
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] == 0) {
                    bRow[i] = true;
                    bCol[j] = true;
                }
            }
        }
        
        for (int i = 0; i < m; i ++) {
            for (int j = 0; j < n; j ++) {
                if (bRow[i] || bCol[j]) {
                    matrix[i][j] = 0;
                }
            }
        }
        
        return;
    }