矩阵置零

矩阵置零

方法一:使用额外的存储空间

• 遍历原始矩阵，如果当前遍历值为0
• 将当前行号i存入rSet集合中去，将当前列号存入cSet集合中去
• 重新遍历一次矩阵，对每个位置检查i,j是否出现在rSet或cSet,只要一个为true,就将当前位置的值置为0

代码实现

class Solution {
    public void setZeroes(int[][] matrix) {
        if(matrix == null || matrix.length == 0){
            return;
        }
        int rows = matrix.length;
        int cols = matrix[0].length;
        Set<Integer> rSet = new HashSet<>();
        Set<Integer> cSet = new HashSet<>();
        for(int i = 0; i < rows; i++){
            for(int j = 0; j < cols; j++){
                if(matrix[i][j] == 0){
                    rSet.add(i);
                    cSet.add(j);
                }
            }
        }
        for(int i = 0; i < rows; i ++){
            for(int j = 0; j < cols; j++){
                if(rSet.contains(i) || cSet.contains(j)){
                    matrix[i][j] = 0;
                }
            }
        }
    }
}

复杂度分析

• 时间复杂度:
• 空间复杂度:

方法二

在方法一中，额外利用了两个集合来记录需要置为0的行和列，但我们可以利用每行和每列的第一个元素作为标记，用来表示这一行或者这一列是否需要赋零。

算法

• 遍历矩阵，如果cell[i][j] == 0 就将cell[i][0] 和cell[0][j]置为0
• 细节:第一行和第一列的标记都是相同的，都是cell[0][0],所以需要一个额外的变量(isCol)告知第一列是否需要被标记，同时用cell[0][0]继续表示第一行的标记
• 从第二行第二列开始遍历，如果cell[i][0] == 0或者cell[0][j] == 0,则cell[i][j] = 0;
• 检查cell[0][0]是否为0，是的话将第一行全部标记为0
• 再检查isCol是否为true,是的话，第一列全部标记为0

代码实现

class Solution {
    public void setZeroes(int[][] matrix) {
        if(matrix == null || matrix.length == 0){
            return;
        }
        int rows = matrix.length;
        int cols = matrix[0].length;
        boolean isCol = false;
        for(int i = 0; i < rows; i++){
            //第一列是否需要被标记
            if(!isCol && matrix[i][0] == 0 ){
                isCol = true;
            }
            //注意；这里需要从第二列开始处理，因为上面的代码已经处理完了第一列
            for(int j = 1; j < cols; j++){
                if(matrix[i][j] == 0){
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        //从第二行第二列开始遍历
        for(int i = 1; i < rows; i++){
            for(int j = 1; j < cols; j++){
                if(matrix[i][0] == 0 || matrix[0][j] == 0){
                    matrix[i][j] = 0;
                }
            }
        }
        //需要先判断第一行以后再来判断第一列，不然可能第一列中将matrix[0][0]置为了0
        //如果第一行需要被标记
        if(matrix[0][0] == 0){
            for(int j = 0; j < cols; j++){
                matrix[0][j] = 0;
            }
        }
        //如果第一列需要被标记
        if(isCol){
            for(int i = 0; i < rows; i++){
                matrix[i][0] = 0;
            }
        }
        
    }
}