在列表中查找具有给定总和的Python三胞胎
在数字列表中，我们希望找出可以结合的三个元素以得出一定的总和。我们称其为三元组。并且在列表中可以有很多这样的三胞胎。例如，总和10可以由数字1,6,3以及1,5,4生成。在本文中，我们将看到如何从给定的数字列表中找出所有这些三胞胎。
使用范围和温度变量
这是我们将创建临时变量的传统方法。这些变量将保留列表中的元素，并检查它们的总和是否等于所需值。然后它将继续将此类变量累加到最终结果集中。
示例
def SumTriplets(listA, sum):
trpltcnt = 0
res = []
for i in range(0, len(listA) - 1):
s = set() tmp = []
# Adding first element
tmp.append(listA[i])
current_sum = sum - listA[i]
for j in range(i + 1, len(listA)):
if (current_sum - listA[j]) in s:
trpltcnt += 1
# Adding second element
tmp.append(listA[j])
# Adding third element
tmp.append(current_sum - listA[j])
# Appending tuple to the final list
res.append(tuple(tmp))
tmp.pop(2)
tmp.pop(1)
s.add(listA[j])
return res
listA = [11,12,13,14,15,16,17,18,19,20]
print("Required triplets:\n",SumTriplets(listA, 40))输出结果
运行上面的代码给我们以下结果-
Required triplets: [(11, 15, 14), (11, 16, 13), (11, 17, 12), (12, 15, 13)]
示例
from itertools import combinations
listA = [11,12,13,14,15,16,17,18,19,20]
def fsum(val):
return sum(val) == 40
res = list(filter(fsum,list(combinations(listA, 3))))
print("Required triplets:\n",res)输出结果
运行上面的代码给我们以下结果-
Required triplets: [(11, 12, 17), (11, 13, 16), (11, 14, 15), (12, 13, 15)]