如何从列表中获取元素对的总和

Question

我有一份清单lst = [2, 4, 5, 6, 7, 8]

• 如果我在搜索12，我的输出应该是（4,8），（5,7）
• 我的当前输出仅显示（4,8）
• 如果我正在打印报税表，那么列表的背面也会打印（4,8），（5,7）， （8,4）、（7,5）

def find_sum(s, lst):
    indices = {x: i for i, x in enumerate(lst)}
#     print(indices)
    for i, x in enumerate(lst):
        target = s - x
        if target in indices:
            return (lst[i], lst[indices[target]])

    return None

lst = [2, 4, 5, 6, 7, 8]
print(find_sum(12, lst)) 

应为(4,8), (5,7)

Answer 1

当您调用return时，您正在结束函数。这意味着一旦你找到第一对，你的函数就结束了，你再也找不到了

要解决这个问题，您应该在名为good_pairs（或任何您想要的）的函数中添加一个数组。与其写return (lst[i], lst[indices[target]])，不如写good_pairs.append((lst[i], lst[indices[target]]))

最后，只需返回good_pairs列表

然后用你想要的格式打印出来

最终代码：

def find_sum(s, lst):
    s.sort()
    good_pairs = []
    indices = {x: i for i, x in enumerate(lst[:len(lst)//2])}
    # print(indices)
    for i, x in enumerate(lst):
        target = s - x
        if target in indices:
            good_pairs.append((lst[i], lst[indices[target]]))

    return good_pairs

lst = [2, 4, 5, 6, 7, 8]
print(find_sum(12, lst)) 

我首先对数组进行排序，然后仅枚举数组的一半以防止重复。归功于@enzo

希望这有帮助，祝你好运：）