【leetcode刷题笔记】Set Matrix Zeroes

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

---

 

题解：因为题目要求原地算法，所以我们只能利用矩阵第一行和第一列存放置零信息。

1. 首先遍历第一行和第一列，看他们是否需要全部置零，用两个变量first_column_zero和first_row_zero来记录；
2. 遍历矩阵，如果某个位置matrix[i][j]出现了0，就把matrix[i][0]和Matrix[0][j]置零；这一步虽然改变了第一行第一列的元素，但是是正确的，因为这一行对应的位置因为Matrix[i][j]处的0最后的确是要变成0的；
3. 遍历矩阵，如果(i,j)处对应的(i,0)或者(0,j)处为0，那么这个元素就置零；
4. 单独处理第一行第一列，根据第一步的变量first_column_zero和first_row_zero。

代码如下：

 1 public class Solution {

 2     public void setZeroes(int[][] matrix) {

 3         int m = matrix.length;

 4         int n = matrix[0].length;

 5         boolean first_column_zero = false;

 6         boolean first_row_zero = false;

 7         

 8         for(int i = 0;i < m;i ++){

 9             if(matrix[i][0] == 0)

10                 first_column_zero =true;

11         }

12         for(int i = 0;i < n;i ++){

13             if(matrix[0][i] == 0)

14                 first_row_zero = true;

15         }

16         

17         for(int i = 1;i < m;i++){

18             for(int j = 1;j < n;j ++){

19                 if(matrix[i][j]== 0){

20                     matrix[i][0] = 0;

21                     matrix[0][j]= 0; 

22                 }

23             }

24         }

25         

26         for(int i = 1;i < m;i++){

27             for(int j = 1;j < n;j ++){

28                 if(matrix[i][0] == 0 || matrix[0][j]== 0 )

29                     matrix[i][j]= 0; 

30             }

31         }

32         

33         if(first_column_zero){

34             for(int i = 0;i < m;i ++)

35                 matrix[i][0] = 0;

36         }

37         if(first_row_zero){

38             for(int i = 0;i < n;i ++)

39                 matrix[0][i] = 0;

40         }

41     }

42 }