LeetCode-073-矩阵置零

矩阵置零

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

进阶：

• 一个直观的解决方案是使用 O(mn) 的额外空间，但这并不是一个好的解决方案。
• 一个简单的改进方案是使用 O(m + n) 的额外空间，但这仍然不是最好的解决方案。
• 你能想出一个仅使用常量空间的解决方案吗？

示例说明请见LeetCode官网。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/set-matrix-zeroes/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一：穷举法

首先，声明2个List分别为zeroRows和zeroColumns用来记录行和列中存在为0元素的行号和列号，然后2次双重遍历matrix数组，分别获取到存在元素为0的行号和列号，并存到zeroRows和zeroColumns中，最后遍历zeroRows和zeroColumns中的行号和列号，将matrix数组中相应的行和列的所有值都置为0。

说明：这是最笨的方法，待优化。

import java.util.ArrayList;
import java.util.List;

public class LeetCode_073 {
    public static void setZeroes(int[][] matrix) {
        List<Integer> zeroRows = new ArrayList<>();
        List<Integer> zeroColumns = new ArrayList<>();
        // 获取哪些行存在元素为0
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[i].length; j++) {
                if (matrix[i][j] == 0) {
                    zeroRows.add(i);
                    break;
                }
            }
        }

        // 获取哪些列存在元素为0
        for (int i = 0; i < matrix[0].length; i++) {
            for (int j = 0; j < matrix.length; j++) {
                if (matrix[j][i] == 0) {
                    zeroColumns.add(i);
                    break;
                }
            }
        }

        // 将存在元素为0的行全都置为0
        for (int i = 0; i < matrix.length; i++) {
            if (zeroRows.contains(i)) {
                for (int j = 0; j < matrix[i].length; j++) {
                    matrix[i][j] = 0;
                }
            }
        }

        // 将存在元素为0的列全都置为0
        for (int j = 0; j < matrix[0].length; j++) {
            if (zeroColumns.contains(j)) {
                for (int i = 0; i < matrix.length; i++) {
                    matrix[i][j] = 0;
                }
            }
        }
    }

    public static void main(String[] args) {
        int[][] matrix = new int[][]{{1, 1, 1}, {1, 0, 1}, {1, 1, 1}};
        setZeroes(matrix);
        for (int[] ints : matrix) {
            for (int anInt : ints) {
                System.out.print(anInt);
            }
            System.out.println();
        }
    }
}

【每日寄语】 自立更生，艰苦奋斗。