Cracking the coding interview--Q1.7

题目

原文：

Write an algorithm such that if an element in an MxN matrix is 0, its entire row and column is set to 0.

译文：

写一个算法使一个MxN的矩阵中出现0的元素的行和列都设为0.

解答

遍历一遍矩阵，将出现0的元素用两个数组将其行列号存储下来，或开一个行数组row和列数组col，当元素a[i][j]等于0时， 就把row[i]和col[j]置为true，第二次遍历的时候，将对应的行列号置为0即可，下面的两种写法：

class Q1_7{
public static void setZero(int[][] matrix){
//存储元素为0的位置的行列号
int[] row=new int[matrix.length];
int[] column=new int[matrix[0].length];
int k=0,l=0;
for(int i=0;i<matrix.length;i++){
for(int j=0;j<matrix[0].length;j++){
if(matrix[i][j]==0){
row[k++]=i;
column[l++]=j;
}
}
}
for(int i=0;i<row.length;i++){
for(int j=0;j<matrix[0].length;j++){
matrix[row[i]][j]=0;
}
}
for(int i=0;i<column.length;i++){
for(int j=0;j<matrix.length;j++){
matrix[j][column[i]]=0;
}
}
}

//另一种写法
public static void setZeros(int[][] matrix){
int[] row=new int[matrix.length];
int[] column=new int[matrix[0].length];
for(int i=0;i<matrix.length;i++){
for(int j=0;j<matrix[0].length;j++){
if(matrix[i][j]==0){
row[i]=1;
column[j]=1;
}
}
}
for(int i=0;i<matrix.length;i++){
for(int j=0;j<matrix[0].length;j++){
if(row[i]==1||column[j]==1){
matrix[i][j]=0;
}
}
}
}
public static void main(String[] args){
int[][] a={
{1,2,0,4},
{0,3,9,5},
{4,6,8,3}
};
//setZero(a);
setZeros(a);
for(int i=0;i<3;i++){
int j;
for(j=0;j<3;j++)
System.out.printf(a[i][j]+",");
System.out.println(a[i][j]);
}
}
}

如有更好方法，还望交流！

---EOF---