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;
            }
        }
    }
}
相关推荐
追随者永远是胜利者3 分钟前
(LeetCode-Hot100)22. 括号生成
java·算法·leetcode·职场和发展·go
CHANG_THE_WORLD12 分钟前
多维数组传参为什么使用列指针?—— 深度解析
数据结构·算法
元亓亓亓38 分钟前
考研408--数据结构--day12--查找&二叉排序树
数据结构·考研·查找·二叉排序树
追随者永远是胜利者41 分钟前
(LeetCode-Hot100)32. 最长有效括号
java·算法·leetcode·职场和发展·go
lifallen42 分钟前
CDQ 分治 (CDQ Divide and Conquer)
java·数据结构·算法
洛豳枭薰1 小时前
Redis 基础数据结构
数据结构·redis
追随者永远是胜利者1 小时前
(LeetCode-Hot100)31. 下一个排列
java·算法·leetcode·职场和发展·go
ValhallaCoder1 小时前
hot100-二分查找
数据结构·python·算法·二分查找
0 0 01 小时前
【C++】矩阵翻转/n*n的矩阵旋转
c++·线性代数·算法·矩阵
m0_531237171 小时前
C语言-指针,结构体
c语言·数据结构·算法