写点什么

【LeetCode】矩阵置零 Java 题解

用户头像
HQ数字卡
关注
发布于: 2021 年 03 月 21 日

题目

给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。


代码


public class DayCode {    public static void main(String[] args) {        int[][] matrix = {{1,1,1},{1,0,1},{1,1,1}};        System.out.println(Arrays.deepToString(matrix));        new DayCode().setZeroes(matrix);        System.out.println(Arrays.deepToString(matrix));    }
/** * 链接 https://leetcode-cn.com/problems/set-matrix-zeroes/ * 时间复杂度 O(m * n) * 空间复杂度 O(m + n) * @param matrix */ public void setZeroes(int[][] matrix) { int m = matrix.length; int 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; } } } }}
复制代码


总结

  • 今天的题目容易理解,按照题意即可完成代码。

  • 采用标记要替换的行和列,进行 2 次循环即可 AC。

  • 坚持每日一题,加油!


发布于: 2021 年 03 月 21 日阅读数: 8
用户头像

HQ数字卡

关注

还未添加个人签名 2019.09.29 加入

LeetCode,略懂后端的RD

评论

发布
暂无评论
【LeetCode】矩阵置零Java题解