题目描述:
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
解题思路:遍历(m-1)*(n-1)个元素,若matrix[i][j]=0,则使matrix[0][j]=0,matrix[i][0]=0。然后遍历第0行的n-1个元素,若matrix[0][j]=0,则j所在的一列全置为0;遍历第0列的m-1个元素,若matrix[i][0]=0,则i所在的一行全置为0。
关键之所在就是这里的第0行,第0列怎么处理,必须的先记录下第0行或第0列刚开始是否有0,只有刚开始有0存在的时候,最后才可以把第0行或第0列的数置为0。
代码如下:
public void setZeroes(int[][] matrix) {
int m=matrix.length,n=matrix[0].length;
boolean row0=false,col0=false;
if(matrix[0][0]==0){
row0=true; col0=true;
}else{
for(int j=1;j<n;j++){
if(matrix[0][j]==0){
row0=true;break;
}
}
for(int i=1;i<m;i++){
if(matrix[i][0]==0){
col0=true;break;
}
}
}
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;
}
}
}
for(int j=1;j<n;j++){
if(matrix[0][j]==0){
for(int i=1;i<m;i++){
matrix[i][j]=0;
}
}
}
for(int i=1;i<m;i++){
if(matrix[i][0]==0){
for(int j=1;j<n;j++){
matrix[i][j]=0;
}
}
}
if(row0){
for(int j=0;j<n;j++){
matrix[0][j]=0;
}
}
if(col0){
for(int i=0;i<m;i++){
matrix[i][0]=0;
}
}
}