20200408
题目:矩阵置零
给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用算法。
示例:
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
思路 :
- 遍历整个矩阵,如果
matrix[i][j] ==0
,就将第i行和第j列的第一个元素标记。 - 第一行和第一列的标记是相同的,都是
matrix[0][0]
,所以需要有一个额外的变量告知第一列是否被标记,同时用matrix[0][0]
继续表示第一行的标记。 - 然后,从第二行第二列的元素开始遍历,如果第r行或者第c列被标记了,那么就将
matrix[r][c]
设为 0。 - 然后我们检查是否
matrix[0][0] == 0
,如果是则赋值第一行的元素为零。 - 然后检查第一列是否被标记,如果是则赋值第一列的元素为零。
code
class Solution{
public void setZeroes(int[][] matrix){
boolean isCol = false;
int R = matrix.length;
int C = matrix[0].length;
//第一此遍历,进行标位。
for(int i=0;i<R;i++){
//判断第一列是不是有为零的
if(matrix[i][0] == 0){
isCol = true;
}
for(int j=1;j<C;j++){
if(matrix[i][j] == 0){
matrix[0][j] = 0;
matrix[i][0] = 0;
}
}
}
for(int i=1;i<R;i++){
for(int j=1;j<C;j++){
if(matrix[i][0] == 0||matrix[0][j] == 0){
matrix[i][j] = 0;
}
}
}
//对第一行进行判断
if(matrix[0][0] == 0){
for(int j=0;j<C;j++){
matrix[0][j] = 0;
}
}
//对第一列进行判断
if(isCol){
for(int i=0;i<R;i++){
matrix[i][0] = 0;
}
}
}
}
参考
Reference