在列表中查找具有给定总和的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)]