用 Python 编写程序，找出第一对和最后一对的乘积相同的四元组的数量

pythonserver side programmingprogramming更新于 2026/1/19 17:48:17

假设我们有一个名为 nums 的数字列表，其中包含唯一的正数 nums。我们必须从 nums 中找到像 (a, b, c, d) 这样的四元组的数量，使得 a*b = c*d，a、b、c 和 d 都是 nums 的不同元素。

因此，如果输入为 nums = [3, 6, 4, 8]，则输出将为 8，因为四元组是 [[3,8,6,4], [3,8,4,6], [8,3,6,4], [8,3,4,6], [6,4,3,8], [4,6,3,8], [6,4,8,3], [4,6,8,3]]。

为了解决这个问题，我们将遵循以下步骤 −

c := 一个新的映射
n := nums 的大小
对于 i在 0 到 n - 1 的范围内，执行
- 对于 i + 1 到 n - 1 范围内的 j，执行
  - x := nums[i] * nums[j]
  - c[x] := 1 + (如果可用，则为 c[x]，否则为 0)
ret := 0
对于 c 中所有值列表中的每个 x，执行
- ret := ret + x *(x - 1)
返回 ret * 4

示例

让我们看看下面的实现以便更好地理解 −

def solve(nums):
   c = {}
   n = len(nums)
   for i in range(n):
      for j in range(i + 1, n):
         x = nums[i] * nums[j]
         c[x] = c.get(x, 0) + 1
   ret = 0
   for x in c.values():
      ret += x * (x - 1)
   return ret * 4

nums = [3, 6, 4, 8]
print(solve(nums))