检查算术序列合法性的Python代码为何始终返回True？

原代码存在两个致命问题，导致它无法正确检查算术序列，甚至会报错或返回错误结果：

问题分析 #

• 索引越界：当输入列表长度小于3时（比如长度为2），循环中访问lst[i+2]会触发IndexError；如果列表长度为1，循环不会执行，函数无返回值，默认返回None而非预期的True（长度≤2的序列本身就是合法算术序列）。
• 提前返回逻辑错误：return True被放在了for循环的每一次迭代里，导致只要前两个相邻差相等，函数就直接返回True，完全不检查后续元素。比如输入[1,3,5,7,9,10]，原函数只会验证前三个元素的差，直接返回True，但实际最后一个元素破坏了序列规则。

修正后的代码 #

def arithmetic(lst):
    # 长度≤2的序列直接判定为合法算术序列
    if len(lst) <= 2:
        return True
    # 计算初始公差
    diff = lst[1] - lst[0]
    # 遍历所有相邻元素验证公差是否一致
    for i in range(1, len(lst)):
        if lst[i] - lst[i-1] != diff:
            return False
    # 所有元素验证通过，返回True
    return True

修正说明 #

1. 先处理特殊场景：长度≤2的序列无需检查，直接返回True，同时避免了索引越界问题。
2. 预先计算初始公差，再遍历所有相邻元素逐一验证，确保整个序列的差值都统一。
3. 仅当所有元素都通过验证后才返回True，保证了检查的完整性。

内容的提问来源于stack exchange，提问作者Superheroo2