题目
编写一种算法,若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;}}}}
执行结果:


















