LeetCode热题18——矩阵置零（18）

最新推荐文章于 2024-07-11 02:39:05 发布

dy小坚果

最新推荐文章于 2024-07-11 02:39:05 发布

阅读量125

点赞数 3

分类专栏： LeetCode热题100 文章标签： leetcode 矩阵算法

本文链接：https://blog.csdn.net/qq_62734493/article/details/135920980

版权

LeetCode热题100 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

文章介绍了如何在LeetCode上的矩阵置零问题中，使用两种方法优化解决方案：一种是遍历矩阵并标记元素，另一种是利用布尔数组记录行和列中出现的0，仅在需要时置零。展示了两种方法的时间复杂度优化。

摘要由CSDN通过智能技术生成

题目链接：https://leetcode.cn/problems/set-matrix-zeroes/description/?envType=study-plan-v2&envId=top-100-liked

打标记

class Solution {
    public void setZeroes(int[][] matrix) {
        int n = matrix.length;
        int m = matrix[0].length;
        //打标记，如果是0就标记为1
        int[][] cnt = new int[n][m];
        for(int i = 0; i < n; i++){
            for(int j = 0; j < m; j++){
                if(matrix[i][j] == 0){
                    cnt[i][j] = 1;
                }
            }
        }
        //对打上标记的位置操作
        for(int i = 0; i < n; i++){
            for(int j = 0; j < m; j++){
                if(cnt[i][j] == 1){
                    for(int k = 0; k < n; k++){
                        matrix[k][j] = 0;
                    }
                    for(int k = 0; k < m; k++){
                        matrix[i][k] = 0;
                    }
                }
            }
        }
    }
}

//优化
class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length, n = matrix[0].length;
        boolean[] row = new boolean[m];
        boolean[] col = new boolean[n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] == 0) {
                    row[i] = col[j] = true;
                }
            }
        }
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (row[i] || col[j]) {
                    matrix[i][j] = 0;
                }
            }
        }
    }
}