O(n)求解特殊递推序列最大值方法
作者:WeekendFlower
时间:2026-02-01
浏览:0
本文介绍一种高效算法,用于在O(n)时间、O(1)空间内求解由数组a构造的特殊前缀累加序列b的最大元素,避免朴素O(n²)暴力构造的开销。

本文介绍一种高效算法,用于在 O(n) 时间、O(1) 空间内求解由数组 `a` 构造的特殊前缀累加序列 `b` 的最大元素,避免朴素 O(n²) 暴力构造的开销。
该问题的核心在于理解序列 b 的生成规律。观察题中示例:
b[0] = 0 b[1] = b[0] + a[0] b[2] = b[1] + a[0] b[3] = b[2] + a[1] b[4] = b[3] + a[0] b[5] = b[4] + a[1] b[6] = b[5] + a[2] b[7] = b[6] + a[0] ...
可发现 b 是按“轮次(round)”分组构造的:第 i 轮(从 0 开始)新增 i+1 个元素,且每轮内部的增量恰好是 a[0], a[1], ..., a[i] 的前缀——即第 i 轮对应所有长度为 1 到 i+1 的 a 前缀和。
更形式化地,令 S_i = a[0] + a[1] + ... + a[i],P_{i,j} = a[0] + ... + a[j](j ≤ i),则第 i 轮起始于 b_start[i],其生成的元素为:
b_start[i] + P_{i,0},
b_start[i] + P_{i,1},
...,
b_start[i] + P_{i,i}因此,该轮最大值为 b_start[i] + max{P_{i,0}, ..., P_{i,i}}。
关键洞察在于:
- b_start[i] 就是上一轮最后一个 b 值,即 b_start[i] = b_start[i-1] + S_{i-1}(定义 b_start[0] = 0);
- max{P_{i,0}, ..., P_{i,i}} 正是 a 的前 i+1 项的最大前缀和,可在遍历 a 时用 max_sum_a = max(max_sum_a, sum_a) 动态维护;
- 因此,无需显式构造整个 b 数组,只需在单次遍历中同步更新:
- 当前轮起始值 b(即上轮累加和);
- 当前已见前缀和的最大值 max_sum_a;
- 全局最大 b 值 max_b。
以下是清晰、健壮的 O(n)/O(1) 实现:
def find_max_linear(a):
b = max_b = 0 # b: 当前轮起始值(即上轮末尾b值);max_b: 全局最大b
sum_a = max_sum_a = 0 # sum_a: 当前前缀和;max_sum_a: 当前最大前缀和
for x in a:
sum_a += x
max_sum_a = max(max_sum_a, sum_a)
max_b = max(max_b, b + max_sum_a) # 当前轮最大b = 起始b + 最大前缀和
b += sum_a # 更新下一轮起始b值:b += 当前完整前缀和
return max_b✅ 正确性保障:b[0] = 0 已包含(max_b 初始化为 0),每轮贡献被精确捕获;
⚠️ 注意事项:
- 输入 a 可含负数或零,算法天然支持(max_sum_a 可能为负,但 max_b 始终取 max(0, ...) 隐含逻辑,因 b[0]=0 是合法候选);
- 若 a 为空,函数返回 0,符合定义;
- 时间复杂度严格 O(n),空间仅用常数变量。
进阶技巧:借助 itertools.accumulate 可写出函数式风格的一行解(时间仍 O(n),但空间 O(n) 因中间序列生成):
from itertools import accumulate
from operator import add
def find_max_linear_oneliner(a):
acc_a = list(accumulate(a)) # [a0, a0+a1, a0+a1+a2, ...]
prefix_sums = list(accumulate(acc_a)) # 各轮起始b值:[0, a0, a0+(a0+a1), ...]
max_prefixes = list(accumulate(acc_a, max)) # 各轮最大前缀和
return max(0, *map(add, [0] + prefix_sums, [0] + max_prefixes))但生产环境推荐首选清晰的四变量循环版本——它兼具最优复杂度、高可读性与强鲁棒性。
作者最新文章
赤友清理大师
2026-09-16 17:43
南邮光擎智算团队:GaN基Micro-LED光计算芯片从理论到流片的突破
2026-09-08 18:35
多张照片怎么合成PDF文件?三种图片转PDF工具怎么选?
2026-09-03 17:04
Excel转PDF防乱版指南:在线与本地双方案及排版检查
2026-09-03 10:04
多个PDF怎么合并成一个?合并后顺序怎么检查?
2026-09-02 19:54
上一篇:
Win11开启上帝模式方法详解
热门文章
更多
精品专题
更多
Mac软件
更多
WINDOWS
更多


































