73. 矩阵置零

可以选择遍历两遍matrix,一遍记录哪些行哪些列需要置0,一遍置0

O(1)空间复杂度的做法:

首先设置两个boolean型变量记录第0行和第0列是否需要置0

然后用第0行和第0列记录需要置0的行和列

java 复制代码
class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length, n = matrix[0].length;
        boolean row0 = false, col0 = false;
        for (int j = 0; j < n; j++) {
            if (matrix[0][j] == 0) row0 = true;
        }
        for (int i = 0; i < m; i++) {
            if (matrix[i][0] == 0) col0 = true;
        }
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][j] == 0) matrix[0][j] = matrix[i][0] = 0;
            }
        }
        // set 0
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) matrix[i][j] = 0;
            }
        }
        if (row0) {
            for (int j = 0; j < n; j++) {
                matrix[0][j] = 0;
            }
        }
        if (col0) {
            for (int i = 0; i < m; i++) {
                matrix[i][0] = 0;
            }
        }
    }
}
相关推荐
木井巳20 分钟前
【递归算法】单词搜索
java·算法·leetcode·决策树·深度优先
咚咚王者1 小时前
人工智能之RAG工程 第一章 RAG 基础与前置知识
人工智能·算法
handler011 小时前
【算法模板】最小生成树:稠密图选 Prim,稀疏图选 Kruskal
c语言·数据结构·c++·算法
Chase_______2 小时前
LeetCode 2461 & 1423:定长滑窗变体精讲,从 HashMap 判重到正难则反的转化技巧
算法·leetcode·职场和发展
此生决int2 小时前
快速复习之数据结构篇——栈和队列
数据结构·c++
WL_Aurora2 小时前
【每日一题】二分算法
python·算法
昵称小白2 小时前
子串专题部分
数据结构·算法·哈希算法
H_BB2 小时前
第17届蓝桥杯备战历程
c++·算法·职场和发展·蓝桥杯
anew___2 小时前
算法分析与设计课程全算法核心概述|期末复习+知识梳理
算法
daad7772 小时前
记录一次上下文切换次数的统计
服务器·c++·算法