题目描述
矩阵中某个元素为 0,就把它所在行和列全部置 0,要求原地。
matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出 = 中间行和中间列置 0
思路解析
关键不是背模板,而是看懂为什么这题适合矩阵 · 标记。
1. 先检查第一行第一列是否已有 0:先检查第一行第一列是否已有 0
2. 扫描内部格子,遇到 0 就标记行头和列头:扫描内部格子,遇到 0 就标记行头和列头
3. 中间 0 标记 row1 和 col1:中间 0 标记 row1 和 col1
4. 标记阶段不急着清零,避免污染后续扫描:标记阶段不急着清零,避免污染后续扫描
5. 第二遍按标记清内部格子:第二遍按标记清内部格子
6. 被标记的行整行清零:被标记的行整行清零
7. 被标记的列整列清零:被标记的列整列清零
8. 最后处理第一行第一列:最后处理第一行第一列
把这句话记住,下次遇到同类题,就能更快选出方向。
用一个小问题检查自己是不是真的懂了。
参考代码(Python)
Python
1class Solution:
2 def setZeroes(self, matrix):
3 m, n = len(matrix), len(matrix[0])
4 row0 = any(matrix[0][j] == 0 for j in range(n))
5 col0 = any(matrix[i][0] == 0 for i in range(m))
6 for i in range(1, m):
7 for j in range(1, n):
8 if matrix[i][j] == 0:
9 matrix[i][0] = matrix[0][j] = 0
10 for i in range(1, m):
11 for j in range(1, n):
12 if matrix[i][0] == 0 or matrix[0][j] == 0:
13 matrix[i][j] = 0
14 if row0:
15 for j in range(n): matrix[0][j] = 0
16 if col0:
17 for i in range(m): matrix[i][0] = 0复杂度分析
- 时间复杂度:O(mn) —— 每个核心状态按算法要求处理固定次数
- 空间复杂度:O(1) —— 只保存必要的辅助结构或递归栈
套路模板
模板不是死背,而是提醒你写代码前先把状态、转移和边界排好。
Python
1# 矩阵 · 标记 通用检查表
2# 1. 定义状态/指针/容器
3# 2. 每轮只做一个清晰动作
4# 3. 更新答案并处理边界易错点
- 错误写法:一看到 0 就立刻清整行整列 → 正确写法:先标记,后清零(立刻清会制造新的 0 干扰扫描)
- 错误写法:只按样例推代码 → 正确写法:先写清状态含义和边界条件(样例太少,隐藏用例专打边界)
- 错误写法:变量名和动画不一致 → 正确写法:代码变量沿用动画里的核心名字(学习时最怕脑内维护两套概念)
以上文字与上方动画完全同一思路:先看动画建立直觉 → 读文字巩固 → 关掉页面自己默写一遍代码,能写出来才算真的掌握。