其他分享
首页 > 其他分享> > LeetCode-073-矩阵置零

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();
        }
    }
}

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

标签:zeroColumns,matrix,int,矩阵,++,length,zeroRows,LeetCode,073
来源: https://www.cnblogs.com/kaesar/p/15164787.html