力扣题解（73. 矩阵置零），带注释

题目描述

链接:点我

题解

//法一  使用hashset记录有0的横纵坐标即可
class Solution {public void setZeroes(int[][] matrix) {HashSet<Integer> row = new HashSet<Integer>();HashSet<Integer> col = new HashSet<Integer>();for(int i = 0;i < matrix.length;i++){for(int j = 0;j < matrix[i].length;j++){if(matrix[i][j] == 0){row.add(i);col.add(j);}}}//更新for(int i = 0;i < matrix.length;i++){for(int j = 0;j < matrix[i].length;j++){if(row.contains(i) || col.contains(j)){matrix[i][j] = 0;}}}}
}//法二，我也是看了题解才明白orz.......  此法的空间复杂度更小O(1)
class Solution {public void setZeroes(int[][] matrix) {int m = matrix.length , n = matrix[0].length;int flag = 0;  //标记第0列是否有0，有为1for(int i = 0;i < m;i++){if(matrix[i][0] == 0) flag = 1;for(int j = 1;j < n;j++){//通过其他列更新第0列，其他行更新第0行if(matrix[i][j] == 0) matrix[i][0] = matrix[0][j] = 0; }}for(int i = m-1;i >= 0;i--){//防止每一列的第一个元素被提前更新，倒着遍历for(int j = 1;j < n;j++){ //注意这里是从1开始if(matrix[i][0] == 0 || matrix[0][j] == 0) matrix[i][j] = 0;}}if(flag == 1){  //更新第0列for(int i =0;i < m; i++) matrix[i][0] = 0;}}
}