力扣 73、矩阵置0

一、问题描述

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

二、原地算法

什么是原地算法，就是空间复杂度很小，是o(1)或者是一个固定的常数。

三、解决思路

1、可以使用矩阵的第一行和第一列来标记这一行或者这一列是否存在0元素，但是要注意的是，我们要保持原来的0行0列是否有0元素，这个时候开辟两个额外的变量来记录一下。如果从0行0列开始标记的话，会出现的一种情况就是可能这个数组的这一行没有0元素，但是由于（0，0）是0，那么，（0，1）也会修改为0；（1，1）也因为（0，1）的修改而错误的改成0。如果用第一行第一列开始标记，不仅能判断第一行第一列是否为0，还能判断本行本列中是否有0。空间复杂度是o(1)。我的描述不是很清楚，自己跟着代码走一遍，就明白了。

2、可以使用两个set来记录，一个记录行标，一个记录列标，空间复杂度是 o(m+n);

三、代码

1、

  public void setZeroes(int[][] matrix) {
        //原地算法，只用o(1)的额外空间，或者很小的固定的空间
        int row = matrix.length;
        int col = matrix[0].length;
        int flag_r = 0;
        int flag_c = 0;
        for (int i = 0; i < row; i++) {
            if (matrix[i][0] == 0)
                flag_r = 1;
        }
        for (int i = 0; i < col; i++) {
            if (matrix[0][i] == 0)
                flag_c = 1;
        }
        for (int i = 1; i < row; i++) {
            for (int j = 1; j < col; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = matrix[0][j] = 0;
                }

            }
        }
        for (int i = 1; i < row; i++) {
            for (int j = 1; j < col; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0)
                    matrix[i][j] = 0;
            }
        }
        if (flag_r == 1) {
            for (int i = 0; i < row; i++) {
                matrix[i][0] = 0;
            }
        }
        if (flag_c == 1) {
            for (int i = 0; i < col; i++) {
                matrix[0][i] = 0;
            }
        }
    }

2、

class Solution {
    public void setZeroes(int[][] matrix) {
        Set<Integer> row_zero = new HashSet<>();
        Set<Integer> col_zero = new HashSet<>();
        int row = matrix.length;
        int col = matrix[0].length;
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if (matrix[i][j] == 0) {
                    row_zero.add(i);
                    col_zero.add(j);
                }
            }
        }
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if (row_zero.contains(i) || col_zero.contains(j)) matrix[i][j] = 0;
            }
        }  
    }
}