「 数据结构与算法 」零矩阵

更多、更体系化的内容请持续关注 《菜鸟程序员成长计划》之数据结构与算法 专栏，您的 关注、点赞、收藏 都将是小编持续创作的动力！

---

题目

编写一种算法，若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的列和行定义为true，后面进行这两个数组的遍历，如果为true时就清除对应矩阵的行、列数据。

class Solution {
    public void setZeroes(int[][] matrix) {
        boolean row[] = new boolean[matrix.length];
        boolean vertical[] = new boolean[matrix[0].length];
        for(int i = 0; i < matrix.length;i++){
            int now[] = matrix[i];
            for (int j = 0; j < now.length; j++) {
                if (now[j] == 0){
                    row[i] = true;
                    vertical[j] = true;
                }
            }
        }

        for (int i = 0; i < row.length; i++) {
            int now[] = new int[matrix[0].length];
            if (row[i]){
                matrix[i] = now;
            }
        }

        for (int i = 0; i < vertical.length; i++) {
            if (vertical[i]){
                for (int j = 0; j < matrix.length; j++) {
                    matrix[j][i] = 0;
                }
            }
        }
    }
}

---

本篇内容参考自开源社区，感谢前人的经验和付出，让我们可以有机会站在巨人的肩膀上眺望星辰大海。