程序员面试金典--题目解析-1.7 若M*N矩阵中某个元素为0，则将其所在的行和列清零

1.7 题目：

编写一个算法，若M*N矩阵中某个元素为0，则将其所在的行和列清零。

解法：

遍历两次矩阵，使用两个布尔数组记录应该被清零的行和列，第一次遍历，记录下需要被清零的。

第二次遍历，清零即可。

1. public static void setZeros(int[][] matrix){  
2.         boolean[] rows = new boolean[matrix.length];  
3.         boolean[] columns = new boolean[matrix[0].length];  
4.           
5.         for(int i=0;i<matrix.length;i++){  
6.             for(int j=0;j<matrix[0].length;j++){  
7.                 if(matrix[i][j] == 0){  
8.                     rows[i] = true;  
9.                     columns[j] = true;  
10.                 }  
11.             }  
12.         }  
13.           
14.         for(int i=0;i<matrix.length;i++){  
15.             for(int j=0;j<matrix[0].length;j++){  
16.                 if(rows[i] || columns[j]){  
17.                     matrix[i][j] = 0;  
18.                 }  
19.             }  
20.         }  
21.     } 

测试用例：

1. @Test  
2.     public void test_1_7(){  
3.         int[][] matrix = {{1,0,3,4},{5,6,7,8},{9,10,11,12},{13,14,15,16}};  
4.         ArrayUtil.setZeros(matrix);  
5.         for(int i=0;i<4;i++){  
6.             for(int j=0;j<4;j++){  
7.                 if(j<3)  
8.                     System.out.print(matrix[i][j]+"\t");  
9.                 else  
10.                     System.out.println(matrix[i][j]);  
11.             }  
12.         }  
13.     } 

测试结果：

0 0 0 0
5 0 7 8
9 0 11 12
13 0 15 16