5-数组-矩阵置零

这是数组的第5篇算法,力扣链接

给定一个 mxn 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法**。**

示例 1:

复制代码
输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出:[[1,0,1],[0,0,0],[1,0,1]]

示例 2:

复制代码
输入:matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出:[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

看到这道题的第一反应就是打表,两次遍历,一次记录清零的行列,最后再遍历清零。

Go 复制代码
func setZeroes(matrix [][]int) {
	rows, cols := make([]bool, len(matrix)), make([]bool, len(matrix[0]))
	for ri, row := range matrix {
		for ci, col := range row {
			if col == 0 {
				rows[ri] = true
				cols[ci] = true
			}
		}
	}
	for ri, row := range matrix {
		for ci, _ := range row {
			if rows[ri] || cols[ci] {
				matrix[ri][ci] = 0
			}
		}
	}
}

这里有多种解法,核心逻辑都差不多,只不过标记的方法不同。

这里延伸出一种特殊的标注逻辑,我们将每一行/列的信息同步到第一行/列中,这样就不会额外占用空间了。值得注意的是,第一列是否需要清零需要额外标注。

Go 复制代码
func setZeroes(matrix [][]int) {
	col0 := false
	for _, row := range matrix {
		if row[0] == 0 {
			col0 = true
		}
		for j := 1; j < len(matrix[0]); j++ {
			if row[j] == 0 {
				row[0] = 0
				matrix[0][j] = 0
			}
		}
	}
	for i := len(matrix) - 1; i >= 0; i-- {
		for j := 1; j < len(matrix[0]); j++ {
			if matrix[i][0] == 0 || matrix[0][j] == 0 {
				matrix[i][j] = 0
			}
		}
		if col0 {
			matrix[i][0] = 0
		}
	}
}
相关推荐
闭着眼睛学算法11 分钟前
【双机位A卷】华为OD笔试之【哈希表】双机位A-跳房子I【Py/Java/C++/C/JS/Go六种语言】【欧弟算法】全网注释最详细分类最全的华子OD真题题解
java·c语言·c++·python·算法·华为od·散列表
自信150413057591 小时前
初学者小白复盘15之指针(4)
c语言·数据结构·算法
郝学胜-神的一滴1 小时前
Cesium绘制线:从基础到高级技巧
前端·javascript·程序人生·线性代数·算法·矩阵·图形渲染
东皇太星2 小时前
机器学习概念,算法原理及应用
算法·机器学习·梯度下降法
.格子衫.2 小时前
021数据结构之并查集——算法备赛
数据结构·算法
im_AMBER2 小时前
Leetcode 35
笔记·学习·算法·leetcode
码农多耕地呗2 小时前
力扣101.对称二叉树(java)
算法·leetcode
小龙报2 小时前
《算法通关指南之C++编程篇(5)----- 条件判断与循环(下)》
c语言·开发语言·c++·算法·visualstudio·学习方法·visual studio
郝学胜-神的一滴2 小时前
C++ STL(标准模板库)深度解析:从基础到实践
linux·服务器·开发语言·c++·算法
刚入坑的新人编程3 小时前
算法训练.17
开发语言·数据结构·c++·算法