LeetCode 面试题 01.08. 零矩阵

Sasakihaise_

于 2022-09-30 19:42:23 发布

阅读量2.9k

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/Sasakihaise_/article/details/127128349

版权

LeetCode 专栏收录该内容

438 篇文章 1 订阅

订阅专栏

这篇博客探讨了如何在Java中实现将二维数组（矩阵）中的所有0元素对应的行和列设置为0的算法。两种方法被提出：一种是通过枚举并使用额外的标记数组，另一种是优化后的方案，仅使用行和列标志数组来节省空间。这两种方法都有效地解决了问题。

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

面试题 01.08. 零矩阵

【枚举】枚举所有位置的数，如果是0，就把当前行和列的元素置为0，但是需要用一个标记来标记这个位置是被置的0，而不是原来就是0。

class Solution {

    // 7:22 9
    // 枚举

    int[][] matrix, visit;
    int m, n;

    public void clear(int x, int y) {
        for (int i = 0; i < m; i++) {
            if (matrix[i][y] != 0) {
                matrix[i][y] = 0;
                visit[i][y] = 1;
            }
        }
        for (int j = 0; j < n; j++) {
            if (matrix[x][j] != 0) {
                matrix[x][j] = 0;
                visit[x][j] = 1;
            }
        }
    }

    public void setZeroes(int[][] matrix) {
        m = matrix.length; n = matrix[0].length;
        this.matrix = matrix;
        this.visit = new int[m][n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (visit[i][j] == 1) continue;
                if (matrix[i][j] == 0) {
                    clear(i, j); 
                }
            }
        }
    }
}

【优化】用一个行数组和一个列数组来标记更省空间。

class Solution {

    // 7:38 4

    public void setZeroes(int[][] matrix) {
        int m = matrix.length, n = matrix[0].length;
        boolean[] row = new boolean[m], 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] = true;
                    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;
            }
        }
    }
}