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;
            }
        }
    }
}
相关推荐
Hag_204 分钟前
LeetCode Hot100 42.接雨水
算法·leetcode·职场和发展
回敲代码的猴子6 分钟前
2月13日打卡
算法
老鼠只爱大米14 分钟前
LeetCode经典算法面试题 #153:寻找旋转排序数组中的最小值(暴力搜索、二分查找等五种实现方案详细解析)
算法·leetcode·二分查找·旋转数组·最小值搜索
春日见22 分钟前
commit与fetch
linux·人工智能·算法·机器学习·自动驾驶
俩娃妈教编程33 分钟前
洛谷选题:P1888 三角函数
c++·算法
TracyCoder12342 分钟前
LeetCode Hot100(56/100)——131. 分割回文串
算法·leetcode
_OP_CHEN42 分钟前
【算法提高篇】(一)线段树之入门篇:从原理到实战,搞定区间操作难题
数据结构·算法·蓝桥杯·线段树·c/c++·区间查询·acm、icpc
IvanCodes43 分钟前
九、C语言动态内存管理
c语言·开发语言·算法
pp起床1 小时前
贪心算法 | part05
算法·贪心算法
MediaTea1 小时前
Python:迭代器的应用场景
开发语言·python·算法