【LeetCode】矩阵置零 Java 题解
题目
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。
代码
复制代码
总结
今天的题目容易理解,按照题意即可完成代码。
采用标记要替换的行和列,进行 2 次循环即可 AC。
坚持每日一题,加油!
版权声明: 本文为 InfoQ 作者【HQ数字卡】的原创文章。
原文链接:【http://xie.infoq.cn/article/de542f32b7b8221f7eda6883b】。文章转载请联系作者。
给定一个 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。
坚持每日一题,加油!
版权声明: 本文为 InfoQ 作者【HQ数字卡】的原创文章。
原文链接:【http://xie.infoq.cn/article/de542f32b7b8221f7eda6883b】。文章转载请联系作者。
还未添加个人签名 2019.09.29 加入
LeetCode,略懂后端的RD
促进软件开发及相关领域知识与创新的传播
评论