力扣HOT100 - 73. 矩阵置零

解题思路：

题目要求使用原地算法

在原地算法中，输入数据通常在内存中直接被修改，而不需要额外的辅助数据结构来存储中间结果。原地算法的一个常见应用是在数组或列表上进行操作，而不是创建新的数组或列表来存储结果。

java 复制代码

class Solution {
    public void setZeroes(int[][] matrix) {
        int row = matrix.length; // 矩阵的行数
        int col = matrix[0].length; // 矩阵的列数
        boolean row0_flag = false; // 标记第一行是否有零
        boolean col0_flag = false; // 标记第一列是否有零

        // 检查第一行是否有零，遍历第一行需要知道有多少列
        for (int j = 0; j < col; j++) {
            if (matrix[0][j] == 0) {
                row0_flag = true;
                break;
            }
        }

        // 检查第一列是否有零，遍历第一列需要知道有多少行
        for (int i = 0; i < row; i++) {
            if (matrix[i][0] == 0) {
                col0_flag = true;
                break;
            }
        }

        // 使用第一行和第一列作为标志位
        for (int i = 1; i < row; i++) {
            for (int j = 1; j < col; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = matrix[0][j] = 0; // 如果元素为零，则将对应的第一行和第一列的元素设置为零
                }
            }
        }

        // 根据第一行和第一列的标志位，将矩阵中的元素置零
        for (int i = 1; i < row; i++) {
            for (int j = 1; j < col; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                    matrix[i][j] = 0; // 如果第一行或第一列的元素为零，则将当前元素置零
                }
            }
        }

        // 如果第一行有零，则将第一行所有元素置零，遍历第一行需要知道有多少列
        if (row0_flag) {
            for (int j = 0; j < col; j++) {
                matrix[0][j] = 0;
            }
        }

        // 如果第一列有零，则将第一列所有元素置零，遍历第一列需要知道有多少行
        if (col0_flag) {
            for (int i = 0; i < row; i++) {
                matrix[i][0] = 0;
            }
        } 
    }
}