【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。

代码如下：

 public class Solution {
     public void setZeroes(int[][] matrix) {
         int m = matrix.length;
         int n = matrix[0].length;
         boolean first_column_zero = false;
         boolean first_row_zero = false;

         for(int i = 0;i < m;i ++){
             if(matrix[i][0] == 0)
                 first_column_zero =true;
         }
         for(int i = 0;i < n;i ++){
             if(matrix[0][i] == 0)
                 first_row_zero = true;
         }

         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 i = 1;i < m;i++){
             for(int j = 1;j < n;j ++){
                 if(matrix[i][0] == 0 || matrix[0][j]== 0 )
                     matrix[i][j]= 0;
             }
         }

         if(first_column_zero){
             for(int i = 0;i < m;i ++)
                 matrix[i][0] = 0;
         }
         if(first_row_zero){
             for(int i = 0;i < n;i ++)
                 matrix[0][i] = 0;
         }
     }
 }