在 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