73. 矩阵置零

中等 含交互动画

学完这道你应该能: 说清它为什么用「标记」,而不是死记步骤; 合上页面,自己默写出核心代码; 用 30 秒把思路讲清楚。

AI 私教 · 就着本题动画讲
卡住了不用硬扛,按 8 步把这题真正走通。
从读题、试解、找法到手写代码,边问边打勾,适合第一次系统学算法的同学。

题目描述

矩阵中某个元素为 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 干扰扫描)
  • 错误写法:只按样例推代码正确写法:先写清状态含义和边界条件(样例太少,隐藏用例专打边界)
  • 错误写法:变量名和动画不一致正确写法:代码变量沿用动画里的核心名字(学习时最怕脑内维护两套概念)

以上文字与上方动画完全同一思路:先看动画建立直觉 → 读文字巩固 → 关掉页面自己默写一遍代码,能写出来才算真的掌握

下一题 →240. 搜索二维矩阵 II ← 返回题库