给定一个 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){//设置对应的第一行第一列对应的位置值为0matrix[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;}}}
}