Leetcode 73 矩阵置零

题目信息

LeetCode地址: . - 力扣(LeetCode)

题目理解

矩阵是m*n的,如果某个元素(i,j)等于0,则将第i行和第j列的所有元素都置零。既然如此,我们可以便利每一个元素,并记录下哪一行哪一列有零。记录完毕后,重新遍历整个矩阵,如果其所在行或者所在列存在零,则置为零。

所以问题变成了,在哪里保存这些信息: 哪一行哪一列有零

我们可以通过一个m长和n长的数组来保存这些信息,这会产生O(m+n)的额外空间复杂度。

我们也可以把这些信息直接保存在矩阵的第一列和第一行!代价是我们要提前遍历第一行和第一列,检查其是否存在零,如果是,最后我们需要把第一列或者第一行的所有元素都置零。

O(m+n)额外空间复杂度写法

java 复制代码
public void setZeroes(int[][] matrix) {
        int[] row = new int[matrix.length];
        int[] col = new int[matrix[0].length];
        for (int i = 0; i<matrix.length; i++) {
            for (int j = 0; j<matrix[0].length; j++) {
                if (matrix[i][j] == 0) {
                    row[i] = 1;
                    col[j] = 1; 
                }
            }
        }
        for (int i = 0; i<matrix.length; i++) {
            for (int j = 0; j<matrix[0].length; j++) {
                if (row[i] == 1 || col[j] == 1) {
                    matrix[i][j] = 0;
                }
            }
        }
    }

O(1)额外空间复杂度写法

java 复制代码
public void setZeroes(int[][] matrix) {
        boolean firstRow = false;
        boolean firstCol = false;
        for (int i = 0; i<matrix.length; i++) {
            if (matrix[i][0] == 0) {
                firstCol = true;
                break;
            }
        }
        for (int j = 0; j<matrix[0].length; j++) {
            if (matrix[0][j] == 0) {
                firstRow = true;
                break;
            }
        }
        for (int i=1;i < matrix.length; i++) {
            for (int j = 1; j<matrix[0].length; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        for (int i =1; i<matrix.length; i++) {
            for (int j = 1; j<matrix[0].length; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                    matrix[i][j] = 0;
                }
            }
        }
        if (firstCol) {
            for (int i =0; i<matrix.length; i++) {
                matrix[i][0] = 0;
            } 
        }
        if (firstRow) {
            for (int i =0; i<matrix[0].length; i++) {
                matrix[0][i] = 0;
            } 
        }
    }
相关推荐
yiridancan12 分钟前
终极剖析HashMap:数据结构、哈希冲突与解决方案全解
java·数据结构·算法·哈希算法
满分观察网友z14 分钟前
性能优化大作战:从 O(N*M) 到 O(N),我的哈希表奇遇记(1865. 找出和为指定值的下标对)
算法
点云SLAM2 小时前
二叉树算法详解和C++代码示例
数据结构·c++·算法·红黑树·二叉树算法
今天背单词了吗9809 小时前
算法学习笔记:19.牛顿迭代法——从原理到实战,涵盖 LeetCode 与考研 408 例题
笔记·学习·算法·牛顿迭代法
jdlxx_dongfangxing10 小时前
进制转换算法详解及应用
算法
why技术11 小时前
也是出息了,业务代码里面也用上算法了。
java·后端·算法
2501_9228955811 小时前
字符函数和字符串函数(下)- 暴力匹配算法
算法
IT信息技术学习圈12 小时前
算法核心知识复习:排序算法对比 + 递归与递推深度解析(根据GESP四级题目总结)
算法·排序算法
愚润求学13 小时前
【动态规划】01背包问题
c++·算法·leetcode·动态规划
会唱歌的小黄李13 小时前
【算法】贪心算法入门
算法·贪心算法