列表索引循环遍历求和函数实现及索引问题求助

问题需求 #

给定整数列表 L = [1, 2, 3, 4, 5] 和数字 x，需实现函数完成以下逻辑：

• 当 x > 0 时：对列表中索引为 j 的元素，累加**索引从 j+1 到 j+x**的元素（索引超出列表长度时循环遍历），结果作为新列表 R 的对应元素。
• 当 x < 0 时：反向累加**索引从 j-1 到 j-x**的元素（同样支持循环遍历）。

示例（当 x=2 时）：

R = [5 (2+3), 7 (3+4), 9 (4+5), 6 (5+1), 3 (1+2)]

原代码存在的问题 #

你编写的 x>0 分支代码有几个核心问题：

1. 使用 L.index(j) 获取索引不可靠：如果列表中有重复元素，index() 只会返回第一个匹配项的索引，导致后续元素处理错误。
2. 索引计算逻辑错误：通过 len(L) 偏移的方式处理循环索引，会导致切片 L[i1:i2] 在跨边界时返回空列表，无法正确累加循环部分的元素。
3. 列表赋值错误：初始化的 R 是空列表，直接通过 R[L.index(j)] 赋值会触发索引越界异常。

修复后的完整实现 #

def decode(L, x):
    n = len(L)
    R = [0] * n  # 预先初始化结果列表，避免索引越界
    
    if x > 0:
        for j in range(n):
            total = 0
            # 遍历j+1到j+x的索引，通过取模实现循环
            for k in range(j+1, j+x+1):
                idx = k % n
                total += L[idx]
            R[j] = total
    elif x < 0:
        abs_x = -x
        for j in range(n):
            total = 0
            # 反向遍历j-1到j-abs_x的索引，取模处理循环
            for k in range(j-1, j-abs_x-1, -1):
                idx = k % n
                total += L[idx]
            R[j] = total
    else:
        # x=0时，累加0个元素，结果全为0（可根据需求调整）
        pass
    
    return R

关键优化点 #

• 直接遍历索引而非元素：用 range(n) 遍历索引 j，避免重复元素导致的索引错误。
• 取模运算处理循环索引：通过 k % n 将超出列表范围的索引映射到合法区间，完美实现循环遍历。
• 预先初始化结果列表：用 [0] * n 创建固定长度的列表，避免赋值时的索引越界问题。
• 兼容x<0的反向逻辑：通过反向步长的 range 遍历，结合取模处理反向循环的索引。

测试验证 #

当 L = [1,2,3,4,5] 且 x=2 时，调用 decode(L,2) 会返回 [5,7,9,6,3]，与示例结果一致。
当 x=-2 时，返回 [3,6,9,7,5]（对应累加前两个反向循环元素：1的反向累加是5+1，2的反向累加是1+2，以此类推）。

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