Pythonで最初と最後のペアの積が同じである4倍の数を見つけるプログラム

numsと呼ばれる数のリストがあり、一意の正の数numsがあるとします。 a * b =c * d、a、b、c、dがすべてnumsの別個の要素であるように、numsから（a、b、c、d）のような4倍の数を見つける必要があります。

したがって、入力がnums =[3、6、4、8]の場合、4倍は[[3,8,6,4]、[3,8,4,6]であるため、出力は8になります。 、[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のサイズ
• 0からn-1の範囲のiの場合、do
  • i +1からn-1の範囲のjの場合、do
    • x：=nums [i] * nums [j]
    • c [x]：=1 +（利用可能な場合はc [x]、それ以外の場合は0）
• ret：=0
• cのすべての値のリストにあるxごとに、
  • ret：=ret + x *（x-1）
• return 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))

入力

[3, 6, 4, 8]

出力

8