Set Matrix Zeroes

思路:先遍历一遍数组,用布尔数组记录下有0元素的行和列,然后再遍历数组,利用布尔数组对对应元素置0。

题目要求用O(m+n)的空间复杂度,所以不能用二维布尔数组,应用两个一维布尔数组,一个记录行出现0的位置,一个记录列出现0的位置。

public void setZeroes(int[][] matrix) {
        int rowsize = matrix.length;
        int columnsize = 0;
        if(rowsize > 0)columnsize = matrix[0].length;
        
        boolean row[] = new boolean[rowsize];
        boolean column[] = new boolean[columnsize];
        
        for(int i = 0;i<rowsize;i++){
            for(int j = 0;j<columnsize;j++){
                if(matrix[i][j] == 0){
                    row[i] = true;
                    column[j] = true;
                }
            }
        }
        
        for(int i = 0;i<rowsize;i++){
            for(int j = 0;j<columnsize;j++){
                if(row[i] || column[j])matrix[i][j]=0;
            }
        }
    }


你可能感兴趣的:(Set Matrix Zeroes)