面试题 01.08. 零矩阵

最新推荐文章于 2024-09-14 18:52:56 发布

Duck_Duck_

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量146

点赞数

分类专栏： Leetcode Java 文章标签： leetcode

本文链接：https://blog.csdn.net/Duck_Duck_/article/details/108239228

版权

Leetcode 同时被 2 个专栏收录

218 篇文章 0 订阅

订阅专栏

Java

215 篇文章 0 订阅

订阅专栏

难度：中等

编写一种算法，若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]
]

分析：

先遍历一遍，记录零元素所在的位置，分行和列，值为true表示该行或该列有零元素；

第二次遍历，将有零元素的行或者列中的所有元素全部置零；

代码：

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

结果：