零矩阵 | leetcode算法学习

刺杀曲奇兔

已于 2022-09-30 16:39:22 修改

阅读量227

点赞数

分类专栏： leetcode 数据结构与算法文章标签：算法 leetcode 矩阵

于 2022-09-30 16:31:12 首次发布

本文链接：https://blog.youkuaiyun.com/qq_53548177/article/details/127125626

版权

数据结构与算法同时被 2 个专栏收录

30 篇文章

订阅专栏

leetcode

21 篇文章

订阅专栏

该博客介绍了一个算法，用于处理二维矩阵。当矩阵中的某个元素为0时，该算法会将该元素所在的整行和整列都置为0。通过遍历矩阵，记录0的位置，然后再次遍历矩阵，根据记录的位置将对应行和列的元素设为0。这种方法避免了直接修改矩阵导致的遗漏问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。

示例 1：
输入：
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出：
[
[1,0,1],
[0,0,0],
[1,0,1]
]

示例 2：
输入：
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出：
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]

这道题的思路很简单，就是找出0所在行列的下标然后输出的时候进行判断，由于直接进行清除会导致同行列的0对应行列不能被清除，所以要先找到所有的0。个人认为妙点在 row[i] = true，col[j] = true这两行，这样不用每次都计算当前数是不是在已记录的下标当中。

    public void setZeroes(int[][] matrix) {
        int rows = matrix.length;
        int columns = matrix[0].length;
        
        boolean[] row = new boolean[rows], col = new boolean[columns];

        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < columns; j++) {
                if (matrix[i][j] == 0){
                    row[i] = true;
                    col[j] = true;
                }
            }
        }
        for (int i = 0; i < rows; i++) {
            for (int j = 0; j < columns; j++) {
                if (row[i] || col[j])
                    matrix[i][j] = 0;
            }
        }
    }