标题：O(n) 时间复杂度求解特殊递推序列的最大值

碧海醫心 2026-01-21 00:00:00 次阅读

本文介绍一种高效算法，用于在 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))