18.矩阵置零（原地算法）

javascript 复制代码

var setZeroes = function (matrix) {
    const rows = matrix.length;
    const cols = matrix[0].length;

    // 标记第一行和第一列是否本身包含0（用于最后处理首行首列）
    let firstRowHasZero = false;
    let firstColHasZero = false;

    // 第一步：检查第一行是否有0
    for (let j = 0; j < cols; j++) {
        if (matrix[0][j] === 0) {
            firstRowHasZero = true;
            break;
        }
    }

    // 第二步：检查第一列是否有0
    for (let i = 0; i < rows; i++) {
        if (matrix[i][0] === 0) {
            firstColHasZero = true;
            break;
        }
    }

    // 第三步：用第一行和第一列标记其他行和列是否需要置零
    for (let i = 1; i < rows; i++) { // 从第1行开始（跳过首行）
        for (let j = 1; j < cols; j++) { // 从第1列开始（跳过首列）
            if (matrix[i][j] === 0) {
                // 用首行对应列标记该列需要置零
                matrix[0][j] = 0;
                // 用首列对应行标记该行列需要置零
                matrix[i][0] = 0;
            }
        }
    }

    // 第四步：根据首行首列的标记，将其他行和列置零
    for (let i = 1; i < rows; i++) {
        for (let j = 1; j < cols; j++) {
            // 如果当前行的首列是0，或当前列的首行是0，则置零
            if (matrix[i][0] === 0 || matrix[0][j] === 0) {
                matrix[i][j] = 0;
            }
        }
    }

    // 第五步：处理第一行（如果本身有0，整行置零）
    if (firstRowHasZero) {
        for (let j = 0; j < cols; j++) {
            matrix[0][j] = 0;
        }
    }

    // 第六步：处理第一列（如果本身有0，整列置零）
    if (firstColHasZero) {
        for (let i = 0; i < rows; i++) {
            matrix[i][0] = 0;
        }
    }
};

解题思路

将第一行和第一列拿出来做标记，然后遍历剩余的矩阵，如果该行或该列的第一个标记为0，就把该值赋值为0，空间复杂度为O(1)

详细解法

1.检查第一行和第一列是否有0

2.用第一行和第一列标记其他行和列是否需要置零，如果该元素为0，就标记该行和该列第一个元素为0

3.根据首行首列为0的元素将其他元素也变为0

4.处理首行和首列，如果原本就有0，将整行或整列变为0