LeetCode热题100--73.矩阵置零--中等

元亓亓亓2025-05-08 10:16

1. 题目

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用原地算法。

示例 1：

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]

输出：[[1,0,1],[0,0,0],[1,0,1]]

示例 2：

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]

输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

2. 题解

java 复制代码

class Solution {
    public void setZeroes(int[][] matrix) {
        int n = matrix.length;
        int m = matrix[0].length;
        boolean[] zeroInRow = new boolean[n];
        boolean[] zeroInLine = new boolean[m];
        for(int i = 0; i < n; i++)
            for(int j = 0; j < m; j++)
                if(matrix[i][j] == 0){
                    zeroInRow[i] = true;
                    zeroInLine[j] = true;
                }
        for(int i = 0; i < n; i++)
            for(int j = 0; j < m; j++)
                if(zeroInRow[i] || zeroInLine[j]){
                    matrix[i][j] = 0;
                }
    }
}

3. 解析

官方题解：矩阵置零

class Solution {public void setZeroes(int[][] matrix)
这定义了一个名为"Solution"的新类，并声明了setZeroes()方法来处理这个任务。
int n = matrix.length;
int m = matrix[0].length;
boolean[] zeroInRow = new boolean[n];
boolean[] zeroInLine = new boolean[m];
这初始化了一些变量，用于存储行和列的信息。zeroInRow是一个布尔数组，如果对应行的元素为零，则它在该索引位置上将被设置为true。同样地，zeroInLine是一个布尔数组，表示相应列是否包含零。
for(int i = 0; i < n; i++)
for(int j = 0; j < m; j++)
if(matrix[i][j] == 0){
zeroInRow[i] = true;
zeroInLine[j] = true;
}
这两个嵌套循环遍历整个矩阵。如果找到一个零元素，则将相应的zeroInRow[i]和zeroInLine[j]设置为true。
for(int i = 0; i < n; i++)
for(int j = 0; j < m; j++)
if(zeroInRow[i] || zeroInLine[j]){
matrix[i][j] = 0;
}
这两个嵌套循环遍历整个矩阵。如果对应的zeroInRow[i]或zeroInLine[j]为true，则将该位置上的元素设置为零。