赞
踩
代码如下:
class Solution { public void setZeroes(int[][] matrix) { if(matrix==null||matrix.length<=0){ return; } int rows = matrix.length; int columns = matrix[0].length; boolean zerox = false; boolean zeroy = false; int[] sigx = matrix[0]; for(int i = 0;i < sigx.length;i++){ if(sigx[i] == 0 ){ zerox = true; } } for(int i = 0;i < rows;i++){ if(matrix[i][0] == 0){ zeroy = true; } } for(int i = 1;i < rows;i++){ for(int j = 1 ; j < columns;j++ ){ int target = matrix[i][j]; if(target==0){ matrix[0][j] = 0; matrix[i][0] = 0; } } } for(int i = 1;i < rows;i++){ if(matrix[i][0]==0){ for(int j = 1;j <columns;j++){ matrix[i][j] = 0; } } } for(int j = 1;j < columns;j++){ if(matrix[0][j] == 0){ for(int i = 1;i < rows;i++){ matrix[i][j] = 0; } } } if(zerox){ for(int j = 0;j < columns;j++){ matrix[0][j] = 0; } } if(zeroy){ for(int i = 0;i < rows;i++){ matrix[i][0] = 0; } } } }
代码有点丑陋,思路是用第一行和第一列做标记,比如(2,3)= 0 ,那么我们就把(2,0)=0,(0,3)=0,这样第一行和第一列就分别记录了各自维度为0的情况(这是第一行和第一列的0可能来自两种地方,一种是自己原来就是0,一种是对应的行或列中有值为0,但是效果是一样的,不论哪种情况,都是要将对应的列或行设置为0,所以不影响),但是要考虑,第一行或第一列如果本身存在0如何标记,毕竟(0,0)位置只有一个,所以我们使用另外的boolean来标记(也可以一个用boolean,一个用(0,0))。
总结一下步骤:先遍历第一行和第一列,进行0标记(标记1),然后对除去第一行和第一列的地方进行遍历,并在相应的位置(第一行和第一列)进行标记(标记2),最后依据标记1和标记2对矩阵的所有位置进行标记。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。