【算法题解】矩阵置零

题目

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。

分析

这个题目并不复杂，只需要遍历一遍数组找出为0的元素，并记录其所在的行、列。最后把这些行、列全部置为零即可。所以可以使用两个数组来分别记录行列是否存在为0的情况。时间复杂度O(m*n)，空间复杂度O(m+n)。具体代码如下：

class Solution {
    public void setZeroes(int[][] matrix) {
        boolean[] row = new boolean[matrix.length];
        boolean[] col = new boolean[matrix[0].length];
        for(int i = 0; i < matrix.length; i++) {
            for(int j = 0; j < matrix[i].length; j++) {
                if(matrix[i][j] == 0) {
                    row[i] = true;
                    col[j] = true;
                }
            }
        }
        for(int i = 0; i < row.length; i++) {
            if(row[i]) {
                for(int j = 0; j < matrix[i].length; j++) {
                    matrix[i][j] = 0;
                }
            }
        }

        for(int j = 0; j < col.length; j++) {
            if(col[j]) {
                for(int i = 0; i < matrix.length; i++) {
                    matrix[i][j] = 0;
                }
            }
        }
    }

}

但是题目要求使用原地算法，原地算法的意思是不使用额外空间。那么我们可以使用第一行、第一列进行记录，为0表示对应行或列存在为0的元素，当第一行、列本身元素就是0时也可以兼容。同时使用两个额外变量记录第一行、第一列原本是否存在0。具体代码如下：

class Solution {
    public void setZeroes(int[][] matrix) {
       boolean flagRow = false,flagCol = false;
       for(int i = 0; i < matrix.length; i++) {
           for(int j = 0; j < matrix[0].length; j++) {
               if(matrix[i][j] == 0) {
                   if(i == 0) {
                       flagRow = true;
                   }
                   if(j == 0) {
                       flagCol = true;
                   }
                   matrix[i][0] = 0;
                   matrix[0][j] = 0;
               }
           }
       }
       for(int i = 1; i < matrix.length; i++) {
           if(matrix[i][0] == 0) {
               for(int j = 1; j < matrix[i].length; j++) {
                   matrix[i][j] = 0;
               }
           }
       }
       for(int j = 1; j < matrix[0].length; j++) {
           if(matrix[0][j] == 0) {
               for(int i = 1; i < matrix.length; i++) {
                   matrix[i][j] = 0;
               }
           }
       }
       if(flagRow) {
           for(int j = 0; j < matrix[0].length; j++) {
               matrix[0][j] = 0;
           }
       }
       if(flagCol) {
           for(int i = 0; i < matrix.length; i++) {
               matrix[i][0] = 0;
           }
       }
    }

}