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

相关推荐
四千岁4 分钟前
Ollama+OpenWebUI 最佳组合:本地大模型可视化交互方案
前端·javascript·后端
写不来代码的草莓熊6 分钟前
el-date-picker ,自定义输入数字自动转换显示yyyy-mm-dd HH:mm:ss格式
前端·javascript·vue.js
wfbcg8 分钟前
每日算法练习:LeetCode 76. 最小覆盖子串 ✅
算法·leetcode·职场和发展
Wect12 分钟前
LeetCode 149. 直线上最多的点数:题解深度剖析
前端·算法·typescript
Wect12 分钟前
JS手撕:手写Koa中间件与Promise核心特性
前端·javascript·面试
qianpeng89715 分钟前
运动声源的到达结构仿真
算法
费曼学习法20 分钟前
线段树:区间查询的"终极武器",一文看透高效范围统计
算法
张元清20 分钟前
React 文件处理:上传、拖放区与对象 URL
前端·javascript·面试
wayz1127 分钟前
Day 2:线性回归原理与正则化
算法·机器学习·数据分析·回归·线性回归
QQ6765800838 分钟前
基于yolo26算法的水下目标检测图像数据集 海洋生物识别 海胆识别 海龟识别数据集 海洋生物监测与保护工作 潜水作业安全辅助系 水下环境感知第10408期
算法·目标检测·水下目标检测·海洋生物识别·海胆 海龟识别·海洋生物监测与保护工作·潜水作业安全辅助 水下环境感知