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
		}
	}
}
相关推荐
_dindong28 分钟前
牛客101:链表
数据结构·c++·笔记·学习·算法·链表
wuk99833 分钟前
基于位置式PID算法调节PWM占空比实现电机转速控制
单片机·嵌入式硬件·算法
派大星爱吃猫41 分钟前
堆的概念、结构与应用详解
c语言·数据结构·
不到满级不改名1 小时前
EM算法 & 隐马尔可夫模型
算法
workflower5 小时前
单元测试-例子
java·开发语言·算法·django·个人开发·结对编程
MicroTech20257 小时前
微算法科技(MLGO)研发突破性低复杂度CFG算法,成功缓解边缘分裂学习中的掉队者问题
科技·学习·算法
墨染点香8 小时前
LeetCode 刷题【126. 单词接龙 II】
算法·leetcode·职场和发展
aloha_7898 小时前
力扣hot100做题整理91-100
数据结构·算法·leetcode
Tiny番茄8 小时前
31.下一个排列
数据结构·python·算法·leetcode
挂科是不可能出现的8 小时前
最长连续序列
数据结构·c++·算法