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