每日一题：矩阵置零

给定一个 mxn 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用原地算法**。**

示例 1：

复制代码

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]

使用两个标记变量。

class Solution {
    public void setZeroes(int[][] matrix) {
boolean row = false;
        boolean col = false;
        int n = matrix[0].length;
        for(int i = 0;i < n;i++){
            if(matrix[0][i] == 0){
                row = true;
            }
        }

        for(int i = 0;i < matrix.length;i++){
            if(matrix[i][0] == 0){
                col = true;
            }
        }
        for(int i = matrix.length - 1;i > 0 ;i--){
            for(int j = 0;j < n;j++){
                if(matrix[i][j] == 0){
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        for(int i = matrix.length - 1;i > 0 ;i--){
            if(matrix[i][0] == 0){
                for(int j = 1;j < n;j++){
                    matrix[i][j] = 0;
                }
            }
        }
        for(int i = 1;i < n;i++){
            if(matrix[0][i] == 0){
                for(int j = 1;j < matrix.length;j++){
                    matrix[j][i] = 0;
                }
            }
        }
        if(row){
            for(int i = 0;i < n;i++){
                matrix[0][i] = 0;
            }
        }
        if(col){
            for(int i = 0;i < matrix.length;i++){
                matrix[i][0] = 0;
            }
        }
        for(int i = 0;i < matrix.length;i++){
            for(int j = 0;j < n;j++){
                System.out.print(matrix[i][j] + " ");
            }
            System.out.print("\n");
        }
    }
}

row和col，用于标记是否需要将第一行和第一列设置为0。然后，通过遍历第一行和第一列的元素，判断是否存在0，如果存在则将对应的布尔变量设置为true。

接下来，使用两个嵌套的for循环遍历矩阵中的每个元素。如果某个元素为0，则将该元素所在的行的第一个元素和所在列的第一个元素都设置为0。这样做的目的是在不直接修改原始矩阵的情况下，记录下哪些行和列需要被设置为0。

然后，再次使用两个嵌套的for循环遍历矩阵中的每个元素。如果某个元素所在的行的第一个元素或所在列的第一个元素为0，则将该元素设置为0。这样就能将所有包含0的行和列都设置为0。

最后，根据之前记录的row和col的值，如果需要将第一行或第一列设置为0，则进行相应的操作。

代码优化后：

class Solution {
    public void setZeroes(int[][] matrix) {
boolean row = false;
        boolean col = false;
        int n = matrix[0].length;
        for(int i = 0;i < n;i++){
            if(matrix[0][i] == 0){
                row = true;
            }
        }

        for(int i = 0;i < matrix.length;i++){
            if(matrix[i][0] == 0){
                col = true;
            }
        }
        for(int i = matrix.length - 1;i > 0 ;i--){
            for(int j = 0;j < n;j++){
                if(matrix[i][j] == 0){
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        for(int i = matrix.length - 1;i > 0 ;i--){
            if(matrix[i][0] == 0){
                for(int j = 1;j < n;j++){
                    matrix[i][j] = 0;
                }
            }
        }
        for(int i = 1;i < n;i++){
            if(matrix[0][i] == 0){
                for(int j = 1;j < matrix.length;j++){
                    matrix[j][i] = 0;
                }
            }
        }
        if(row){
            for(int i = 0;i < n;i++){
                matrix[0][i] = 0;
            }
        }
        if(col){
            for(int i = 0;i < matrix.length;i++){
                matrix[i][0] = 0;
            }
        }

    }
}