法(一)一个非常简单,3分钟就能写出来的解法
先遍历一遍矩阵,用boolean[] row[] 和boolean[] col去记录哪些行和哪些列是0.想·
然后再遍历一遍矩阵,置0
class Solution { public void setZeroes(int[][] matrix) { int m = matrix.length; //行 int n = matrix[0].length; //列 boolean[] row = new boolean[m]; boolean[] col = new boolean[n]; //标记哪些行,哪些列是0 for(int i=0;i<m;i++){ for (int j=0; j<n; j++){ if(matrix[i][j]==0){ row[i]=true; col[j]=true; } } } //填充0 for(int i=0;i<m;i++){ for (int j=0; j<n; j++){ if(row[i]||col[j]){ matrix[i][j]=0; } } } } }空间复杂度:O(m+n)
然后,让用O(1)的空间复杂度,给出的方法是,占用原矩阵第一行和第一列用来标记。。。无语了,为了省而省,一点也不优雅。。
但是呢,也有很多要考虑的地方。
(1)考虑既然在原有矩阵基础上标记,那就不能用boolean类型了,只能用int类型,如何标记呢?分析了一下,用0标记就好了。这样最后我们填充0的时候,只要matrix[i][0]==0(或者matrix[0][i]==0),这一行(列)就都标记0。即使是matrix[i][0]本来就是0,那最后填充的时候,这一行也都标记为0,没毛病。
(2)考虑需要一开始对第一行和第一列做清空处理吗,还是什么都不用做。答案是什么都不用做。保留第一行和第一列的原数据,然后遍历去掉第一行和第一列的“小矩形”,去覆盖第一行和第一列的值。
(3)还有就是,不能只管“小矩形”啊。一开始要遍历第一行和第一列,看看第一行和第一列里是否有0,这里会用到额外的空间去记录
class Solution { public void setZeroes(int[][] matrix) { int m = matrix.length; //行 int n = matrix[0].length; //列 boolean firstRow = false; boolean firstCol = false; //记录第一行和第一列是否有0 for(int j=0; j<n; j++){ if(matrix[0][j]==0) firstRow=true; } for (int i=0; i<m; i++){ if(matrix[i][0]==0) firstCol=true; } //标记哪些行,哪些列是0 //注意,遍历的是小矩阵,从i和j都是1开始 for(int i=1;i<m;i++){ for (int j=1; j<n; j++){ if(matrix[i][j]==0){ matrix[i][0]=0; matrix[0][j]=0; } } } //填充0 //注意,填充的是小矩阵,从i和j都是1开始 for(int i=1;i<m;i++){ for (int j=1; j<n; j++){ if(matrix[i][0]==0||matrix[0][j]==0){ matrix[i][j]=0; } } } //最后填充第0行和第0列 if(firstRow){ for(int j=0; j<n; j++){ matrix[0][j]=0; } } if(firstCol){ for (int i=0; i<m; i++){ matrix[i][0]=0; } } } }整个过程有点像穿脱原则。