【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;
        }
    }
}