在 Python 中查找具有相同产品的元组的程序
假设我们有一个具有唯一正值的数组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