提供了编程的基础技术教程

zl程序教程

您现在的位置是：首页 >  其它

当前栏目

零矩阵题解

矩阵 题解
2023-06-13 09:14:29 时间

题目

编写一种算法，若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，这种方式是可以的，但是要注意修改了的行跟列不能参与后面的数据判断，否则后修改的值跟原始数据会产生混淆。所以如果使用这种方式需要标记行列为已修改值。

如果都在一个循环里进行标记和值刷新有点混乱不好梳理，所以将标记跟值刷新分为两步。先定义宽高两个数组把为0的列和行定义为true，后面进行这两个数组的遍历，如果为true时就清除对应矩阵的行、列数据。

代码如下：

    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;
                }
            }
        }
    }

执行结果：

相关文章