在 Python 中查找具有相同产品的元组的程序

• Z时代
• 2024-01-10
• 分类：综合

假设我们有一个具有唯一正值的数组 nums，我们必须找到元组 (a, b, c, d) 的数量，使得 a*b = c*d 其中 a、b、c 和 d 是 nums 的元素，并且所有元素 a、b、c 和 d 都是不同的。

所以，如果输入像 nums = [2,3,4,6]，那么输出将是 8，因为我们可以得到像 (2,6,3,4), (2,6,4,3) 这样的元组, (6,2,3,4), (6,2,4,3), (3,4,2,6), (4,3,2,6), (3,4,6,2) , (4,3,6,2)。

示例

让我们看看以下实现以获得更好的理解 -

from collections import defaultdict
def solve(nums):
   dic = defaultdict(int)
   ans=0
   for i in range(len(nums)-1):
      for j in range(i+1,len(nums)):
         dic[nums[i]*nums[j]]+=1
   for v in dic.values():
      if v==1:
         continue
      v=v-1
      s=(v/2) * (8+8*v)
      ans+=s
   return int(ans)
nums = [3,4,6,2]
print(solve(nums))

输入

[3,4,6,2]

0

以上是 在 Python 中查找具有相同产品的元组的程序 的全部内容， 来源链接： utcz.com/z/343744.html