本文介绍一种基于优先队列(堆)的算法,用于在允许重复选取且必须按序遍历的约束下,准确求出整数数组中所有可能组合中 ≥ 给定 limit 的最小和与最大和,并支持路径还原。
本文介绍一种基于优先队列(堆)的算法,用于在允许重复选取且必须按序遍历的约束下,准确求出整数数组中所有可能组合中 ≥ 给定 limit 的最小和与最大和,并支持路径还原。
该问题的核心约束具有三重特性:
- ✅ 元素可重复使用(如 100 可取多次);
- ✅ 必须按数组顺序选取(即若选了 arr[i],后续只能选 arr[i] 或 arr[i+1],不可跳过中间元素,如 [100,200,300] 中不允许 100+300,但允许 100+200+300 或 100+100+200);
- ✅ 一旦累加和 ≥ limit 即终止当前路径(非必须恰好等于,而是“首次达标”即停)。
这本质上是一个带序约束的无界背包变种问题:目标不是恰好装满,而是在满足单调索引推进的前提下,找到所有可行和中 ≥ limit 的极值(min/max)。传统 DFS 递归易因状态爆炸或剪枝不当导致错误(如示例中添加 10000 后递归法误将 10000 直接计入最大和);而朴素 DP 表设计若未正确建模“索引连续性”与“提前终止”,也会产生偏差(如第二例中动态法输出 1012 而非正确 1014)。
正确解法:基于最小堆的有序状态搜索
我们采用 Dijkstra-like 状态空间搜索,以 (current_sum, next_index) 为状态节点,用最小堆确保优先探索更小的和,同时全程维护最大和候选值:
import heapq
from typing import Tuple, Optional, List
def min_max_limit_sum(arr: List[int], limit: int) -> Tuple[int, List[int], int, List[int]]:
"""
在满足“按序选取(可重复)、首次≥limit即停止”约束下,
返回最小可行和、对应路径、最大可行和、对应路径。
Args:
arr: 严格递增/有序整数数组(题目给定为ordered)
limit: 非负整数阈值
Returns:
(min_sum, min_path, max_sum, max_path)
路径为选取的元素列表,按实际选取顺序排列
"""
if not arr or limit = limit:
if s best_max:
best_max = s
max_path = _reconstruct_path(prev_path, arr, i)
continue # 达标后不向下扩展(题目要求“stop as soon as reach limit”)
# 否则:从索引 i 开始,可选 arr[i](重复)或推进到 arr[i+1]
if i List[int]:
"""从嵌套元组路径结构还原为列表"""
path = []
curr = prev
while curr is not None:
val, curr = curr
path.append(val)
return list(reversed(path))
关键设计解析
- 状态定义精准:(sum, next_index) 明确表示“当前和为 sum,下一步只能从 arr[next_index] 开始选”,天然满足“不可跳过”的约束;
- 双分支扩展:每个状态生成两个子状态——复用当前元素(保持 i 不变)和推进到下一元素(i+1),覆盖所有合法序列;
- 堆驱动最优性:最小堆保证 best_min 第一时间收敛;而 best_max 在遍历中持续更新,因所有达标状态均被枚举(堆虽按 sum 小顶排序,但大和仍会被访问到);
- 路径可追溯:通过 (value, prev_path) 元组链式存储,支持 O(L) 时间还原完整选取序列。
运行验证
# 示例1
arr1, lim1 = [100, 200, 300, 1000], 1000
min_s, min_p, max_s, max_p = min_max_limit_sum(arr1, lim1)
print(f"Min: {min_s} = {min_p}") # Min: 1000 = [100, 100, ..., 100] (10×)
print(f"Max: {max_s} = {max_p}") # Max: 1900 = [100, 200, 300, 300, 1000]
# 示例2
arr2, lim2 = [3, 10, 15], 1000
min_s, min_p, max_s, max_p = min_max_limit_sum(arr2, lim2)
print(f"Max: {max_s} = {sum(max_p)}") # Max: 1014 = 318×3 + 3×10 + 2×15
注意事项与优化建议
- ⚠️ 时间复杂度:最坏为 O(S·N·log(S·N)),其中 S 是可达和的范围。对极大 limit 或密集小数值数组,可考虑数学优化(如 Frobenius 数近似);
- ⚠️ 空间限制:visited 字典可能占用较大内存,若仅需极值无需路径,可简化为 seen = set() 存储 (sum, i);
- ✅ 鲁棒性增强:生产环境应加入 limit 溢出检查、空数组/零值处理及 arr 有序性校验;
- ✅ 扩展性:本框架易于支持更多约束,如“最多使用 k 次某元素”或“路径长度上限”,只需调整状态定义与转移逻辑。
该算法彻底规避了原始递归法的状态遗漏与动态规划法的状态建模缺陷,以清晰的状态语义和高效的堆搜索,确保在多项式可行时间内给出严格正确的最小与最大和解。











