前缀和统计子数组数量
514 字
3 分钟
前缀和统计子数组数量
前缀和统计子数组数量
前缀和适合处理连续区间求和。统计和为 k 的子数组时,关键是把区间和转化成两个前缀和之差。
算法题真正要练的是把题目翻译成状态、边界和转移。模板可以帮助提速,但不能替代对条件的判断。
题型识别
遍历到当前位置时,当前前缀和是 s。如果之前出现过 s-k,说明从那个位置之后到当前位置的区间和是 k。
遇到类似题时,我会先写一句话解释输入、输出和约束,再决定用什么结构。只要这一步说不清,直接写代码大概率会在边界条件上出错。
核心写法
from collections import defaultdict
def subarray_sum(nums, k): count = defaultdict(int) count[0] = 1 s = 0 ans = 0 for x in nums: s += x ans += count[s - k] count[s] += 1 return ans边界样例
- 空输入或只有一个元素时,答案应该是什么。
- 元素重复时,比较关系是否仍然成立。
- 结果需要下标、长度、数量,还是具体路径。
- 时间复杂度能不能从每个元素处理次数解释出来。
复盘
复盘时要先把前缀和 0 放进计数表,表示从数组开头开始的区间也能被统计到。
最小验收样例
我会至少准备三个样例:一个普通样例,一个边界样例,一个专门卡比较条件的样例。算法代码最怕只过了题目给的样例,却没有覆盖空输入、重复元素或极端顺序。
写完后不要只看答案,还要能解释每一步状态为什么变化。只要能把状态变化讲清楚,换一个题面时就不容易机械套错模板。
小结
前缀和统计子数组数量 的练习重点,是把固定套路拆成可解释的判断。能说明为什么这样维护状态,才算真正掌握。
文章分享
如果这篇文章对你有帮助,欢迎分享给更多人!













