LeetCode 73. 矩阵置零
题目描述
🔥 73. 矩阵置零
思路分析
解题思路:
- 遍历矩阵,用两个集合分别记录需要置零的行和列。
- 遍历矩阵,将需要置零的行和列的元素置为 0。
具体步骤:
- 初始化两个集合 rows 和 cols,用于记录需要置零的行和列。
- 遍历矩阵,如果 matrix[i][j] 为 0,则将 i 加入 rows,将 j 加入 cols。
- 遍历矩阵,如果 i 在 rows 中或者 j 在 cols 中,则将 matrix[i][j] 置为 0。
时间复杂度:O(m * n),其中 m 和 n 分别为矩阵的行数和列数。 空间复杂度:O(m + n),其中 m 和 n 分别为矩阵的行数和列数。
参考代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
func setZeroes(matrix [][]int) {
m := len(matrix)
n := len(matrix[0])
rows := make([]bool, m)
cols := make([]bool, n)
// 标记含有零元素的行和列
for i := 0; i < m; i++ {
for j := 0; j < n; j++ {
if matrix[i][j] == 0 {
rows[i] = true
cols[j] = true
}
}
}
// 将相应行和列置为零
for i := 0; i < m; i++ {
for j := 0; j < n; j++ {
if rows[i] || cols[j] {
matrix[i][j] = 0
}
}
}
}
1
write your code here
相似题目
题目 | 难度 | 题解 |
---|---|---|
生命游戏 | Medium |
CC BY-NC-SA 4.0
许可协议,转载请注明出处!
本博客所有文章除特别声明外,均采用