给定一个 m x n
的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
输入:matrix = [[1,1,1],[1,0,1],[1,1,1]] 输出:[[1,0,1],[0,0,0],[1,0,1]]
输入:matrix = [[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对应的行列(非第一行第一列)设置为0。
最后单独处理第一行第一列,根据标志位判断是否应该全置0
class Solution {
public void setZeroes(int[][] matrix) {
boolean firstRowZero = false;
boolean firstColZero = false;
//先遍历第一行
for (int i = 0; i < matrix[0].length; i++) {
if (matrix[0][i] == 0) {
firstRowZero = true;
break;
}
}
//遍历第一列
for (int i = 0; i < matrix.length; i++) {
if (matrix[i][0] == 0) {
firstColZero = true;
break;
}
}
//处理剩余
for(int i = 1; i< matrix.length; i++){
for(int j = 1; j < matrix[0].length; j++){
if(matrix[i][j] == 0){
//设置对应的第一行第一列对应的位置值为0
matrix[0][j] = 0;
matrix[i][0] = 0;
}
}
}
//处理第一行
for(int i = 1; i < matrix[0].length; i++){
if(matrix[0][i] == 0){
for(int j = 1; j < matrix.length; j++){
matrix[j][i] = 0;
}
}
}
//处理第一列
for(int i = 1; i < matrix.length; i++){
if(matrix[i][0] == 0){
for(int j = 1; j < matrix[0].length; j++){
matrix[i][j] = 0;
}
}
}
//处理
if(firstRowZero){
Arrays.fill(matrix[0], 0);
}
if(firstColZero){
for(int i = 0; i < matrix.length; i++){
matrix[i][0] = 0;
}
}
}
}