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

相关推荐
传人once7 分钟前
企业官网静态页基础结构如何搭
javascript·css
桔筐10 分钟前
Vue中DOM的data-*自定义属性(dataset)与Vue data响应式变量区别详解
前端·javascript·vue.js
damoluomu16 分钟前
静态站点生成器怎么选:Node.js 生态里常用的五个(番外一)
javascript·css·typescript·node.js·html
西索斯coding1 小时前
Claude Code 工具调用返回空怎么办?claude-opus-5.5 的 input_schema 校验踩坑记录与修复方案
服务器·前端·javascript·ai
Jlzn88881 小时前
华南PACK线自动化率怎么统计才靠谱
前端·javascript·自动化
鶴哥只手遮天1 小时前
《OpenFOAM 源码深度剖析》系列-第 3 章 字典系统、IO 体系与对象注册表
算法·开源
鶴哥只手遮天1 小时前
《OpenFOAM 源码深度剖析》系列-第 2 章 wmake:OpenFOAM 自研构建系统与编译配置体系
算法·开源
用户81849100742161 小时前
我把工位上的“AI牛马”养成了:一个技术人的 AiPy 实战手记
算法
Tisfy1 小时前
LeetCode 3727.最大交替平方和:排序 (双指针可选)
算法·leetcode·题解·贪心·数组·排序
鶴哥只手遮天1 小时前
《OpenFOAM 源码深度剖析》系列-第 9 章 有限体积离散框架:`fvMatrix`、`fvSchemes` 与 `fvSolution`
算法·开源