本文介绍一种改进的列表求和算法:当遇到6或9时,将二者视为一对“边界”,无论顺序是6→9还是9→6,其间的数字均被排除;而连续的6…6或9…9之间的数字仍被计入总和。
在常规的“跳过6到9之间数字”的求和问题中,通常只需用一个布尔标志(如 between6And9)跟踪是否处于6之后、9之前的状态。但本题的关键“ twist ”在于:6和9构成的是双向可逆的排除区间——即 [6, ..., 9] 和 [9, ..., 6] 都应被跳过,而 [6, ..., 6] 或 [9, ..., 9] 中间的数字必须保留。
这意味着不能仅靠单标志位判断,而需区分两种状态:
更准确地说:每当一个新的6或9出现时,它会“关闭”前一段未闭合的同类型区间,并“开启”一段新类型的待排除区间;而上一段未被关闭的区间(如旧6到新9之间)应被累加进结果。
以下是一个清晰、健壮的实现:
def twistersum(nums):
total = 0
buffer = 0 # 暂存当前待定区间的和(可能被丢弃,也可能被加入total)
saw6 = False # 标记是否已遇到一个未闭合的6
saw9 = False # 标记是否已遇到一个未闭合的9
for x in nums:
if x == 6:
# 遇到6:若之前有未闭合的6(即saw6为True),说明出现了6...6 → 前段buffer有效,应加入total
# 或者之前既无6也无9(saw6==False and saw9==False),说明这是首个边界,buffer清零前无需累加
if saw6:
total += buffer
# 重置缓冲区,标记新6生效,同时解除9状态(因6优先覆盖9的待匹配)
buffer = 0
saw6 = True
saw9 = False
elif x == 9:
# 同理:若之前有未闭合的9(saw9为True),说明出现9...9 → buffer有效,加入total
if saw9:
total += buffer
buffer = 0
saw9 = True
saw6 = False
else:
# 普通数字:累加到当前buffer(它最终是否计入total,取决于后续是否被6/9截断)
buffer += x
# 循环结束后,buffer中剩余的是最后一段未被任何6/9截断的数字之和(例如结尾是...6,2,3 → 但没遇到9就结束了,则2+3不计入)
# 注意:根据规则,只有成对的6↔9才触发排除;末尾悬空的buffer属于“未开启有效排除区间”的部分,应计入
total += buffer
return total✅ 验证示例:
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
? 分段解析该示例:
] → 6到9之间(7,3,5)被跳过 → buffer清零,不加 ⚠️ 关键注意事项:
掌握此模式,不仅能解决本题,还可拓展至多边界、嵌套排除等更复杂的数据过滤场景。