LeetCode73.矩阵置零(Java)

题目

给定一个 m x n 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。(在计算机科学中，一个原地算法（in-place algorithm）是一种使用小的，固定数量的额外之空间来转换资料的算法。当算法执行时，输入的资料通常会被要输出的部分覆盖掉。不是原地算法有时候称为非原地（not-in-place）或不得其所（out-of-place）。)

跳转链接：LeetCode73

解析

由于本题要求使用原地算法，也就是我们不能不能 new 一个新的二维数组，只能在 matrix 上进行修改。但是我们不能直接在第一次遍历的时候修改，因为如果第一次遍历时就开始修改的换，后面就无法知道遍历到的元素在修改前是0还是非0了。所以我们第一次遍历的时候可以对要修改的行和列进行记录，第二次遍历的时候再进行修改。

public class Solution {

    public void setZeroes(int[][] matrix) {
        int n = matrix.length;
        int m = matrix[0].length;
        //用于标记需要修改的行和列
        boolean[] x = new boolean[n + 1];
        boolean[] y = new boolean[m + 1];
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[i].length; j++) {
                if(matrix[i][j] == 0){
                    x[i] = true;
                    y[j] = true;
                }
            }
        }
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[i].length; j++) {
                if(x[i] || y[i]){
                    matrix[i][j] = 0;
                }
            }
        }
    }

}