如何在列表求和中智能跳过6与9之间的所有数字(双向区间,含6→9和9→6)

本文介绍一种改进的列表求和算法:跳过每一对相邻的6与9(或9与6)之间所有数字,但保留6–6、9–9区间内的数字;支持多次出现的6/9,实现真正的“双向排除区间”。

在常规的“跳过6到9之间数字”问题中,我们只需用一个布尔标志 between6And9 跟踪是否处于6之后、9之前即可。但本题的核心挑战在于“双向性”:不仅 6 → ... → 9 区间要被排除,9 → ... → 6 区间同样需要排除;而两个6之间(如 [6,1,2,6])或两个9之间(如 [9,5,9])的数字必须保留

这意味着不能仅靠单标志位判断,而需区分两种独立状态:

  • 当前是否刚遇到一个 6(且尚未匹配到后续的9)?
  • 当前是否刚遇到一个 9(且尚未匹配到后续的6)?

更关键的是:每次新遇到6或9时,应立即结算此前累积的、“安全”的数字和(即上一个有效区间的和),然后重置临时累加器。只有当6与9(或9与6)成对出现时,中间段才被丢弃;否则,数字始终暂存于 temp 中,等待确认是否属于可累加区间。

以下是完整、健壮的实现:

def twistersum(nums):
    the_sum = 0
    temp = 0
    saw6 = saw9 = False  # 分别标记最近一次遇到的是6还是9

    for num in nums:
        if num == 6:
            # 遇到6:若此前已见过6(即6...6),或既没见6也没见9(首个6),则当前temp属于安全段,可加入总和
            if saw6 or (not saw6 and not saw9):
                the_sum += temp
            temp = 0
            saw6 = True
            saw9 = False  # 重置9状态:现在以6为起点,等待匹配9
        elif num == 9:
            # 遇到9:若此前已见过9(即9...9),或既没见6也没见9(首个9),则temp安全
            if saw9 or (not saw6 and not saw9):
                the_sum += temp
            temp = 0
            saw6 = False
            saw9 = True  # 重置6状态:现在以9为起点,等待匹配6
        else:
            # 普通数字:持续累加到temp(暂不决定归属,等边界触发结算)
            temp += num

    # 循环结束后,将最后一段未被边界触发的temp加入总和(如结尾是数字,或结尾是6/9但无配对)
    the_sum += temp
    return the_sum

# 测试用例
list1 = [1, 2, 3, 4, 6, 7, 3, 5, 9, 7, 2, 4, 9, 4, 5, 6, 7, 8, 9, 4, 3, 2, 1]
print(twistersum(list1))  # 输出:33

算法逻辑解析

  • temp 始终缓存“当前待定区间”的数字和;
  • 每次遇到6或9,先将当前 temp 合理结算(依据 saw6/saw9 状态判断该段是否“已完成且安全”),再清空 temp 并更新状态;
  • saw6=True 表示“上一个关键数是6,现正期待9来闭合区间”;同理 saw9=True 表示“正期待6”;
  • saw6 and saw9 的情况不会发生(因每次设True时必设另一个为False),因此状态始终互斥且明确;
  • 结尾的 the_sum += temp 是必须的——它处理末尾无配对边界(如 [..., 5] 或 [..., 6])的情形。

⚠️ 注意事项

  • 该算法不依赖6和9数量相等,能正确处理悬空6或9(如 [6,1,2] → 忽略全部;[1,2,9] → 忽略全部;[6,1,9,2,6,3] → 仅累加最后的3);
  • 时间复杂度 O(n),空间复杂度 O(1),适合大规模数据;
  • 若需调试,可在循环内添加 print(f"num={num}, saw6={saw6}, saw9={saw9}, temp={temp}, sum={the_sum}") 观察状态流转。

通过双状态机设计,我们精准建模了“6↔9双向排除区间”的语义,使代码既简洁又具备强鲁棒性——这才是工程级Python算法应有的表达力。