LeeCode每日一题——2022-09-30

小D带你学

已于 2022-09-30 16:55:25 修改

阅读量114

点赞数

分类专栏： LeeCode 每日一题文章标签：算法

于 2022-09-30 16:39:33 首次发布

本文链接：https://blog.csdn.net/qq_27897465/article/details/127125702

版权

LeeCode 每日一题专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目—零矩阵

编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。

示例 1：

输入：
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出：
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

示例 2：

输入：
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出：
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

个人解答(JAVA)

思路：先循环数据，把等于0的找出来，然后将所对应要置为0的坐标存在一个数组中，然后在循环数据将原二维数组中置为0即可

	public void setZeroes(int[][] matrix) {
        List<String> data = new ArrayList<>();
        for (int i = 0; i < matrix.length; i++) {
            int[] row = matrix[i];
            for (int j = 0; j < row.length; j++) {
                if (matrix[i][j] == 0) {
                    for (int k = 0; k < row.length; k++) {
                        data.add(i + "-" + k);
                    }
                    for (int k = 0; k < matrix.length; k++) {
                        data.add(k + "-" + j);
                    }
                }
            }
        }
        for (String s : data) {
            String[] ss = s.split("-");
            matrix[Integer.parseInt(ss[0])][Integer.parseInt(ss[1])] = 0;
        }
    }

在上面的代码提交后，虽然可以解决问题，但是运行的效率不够好，就有继续优化了下：在初始的时候定义两个数组，用于记录需要置为0的行和列，然和循环二维数组将所需置为0的行和列设置为true，然后再次遍历数组，将行和列均为true的坐标上的值置为0即可，优化后的代码为：

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