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;
            }
        }
    }
}
相关推荐
JYeontu3 分钟前
程序员都该掌握的“质因数分解”
前端·javascript·算法
with-the-flow12 分钟前
从数学底层的底层原理来讲 random 的函数是怎么实现的
c语言·python·算法
tyb33333316 分钟前
leetcode:吃苹果和队列
算法·leetcode·职场和发展
多恩Stone21 分钟前
【3D-AICG 系列-15】Trellis 2 的 O-voxel Shape: Flexible Dual Grid 代码与论文对应
人工智能·python·算法·3d·aigc
weixin_4481199421 分钟前
Datawhale 大模型算法全栈基础篇 202602第4次笔记
笔记·算法
网小鱼的学习笔记22 分钟前
leetcode876:链表的中间结点
数据结构·链表
sali-tec22 分钟前
C# 基于OpenCv的视觉工作流-章27-图像分割
图像处理·人工智能·opencv·算法·计算机视觉
TracyCoder1231 小时前
LeetCode Hot100(60/100)——55. 跳跃游戏
算法·leetcode
月挽清风1 小时前
代码随想录第35天:动态规划
算法·动态规划
岛雨QA1 小时前
链表「Java数据结构与算法学习笔记3」
数据结构·算法