与えられたリストにピタゴラストリプレットがあるかどうかをPythonでチェックするプログラム

numsという数字のリストがあるとすると、a ^ 2 + b ^ 2 =c ^ 2となるような3つの数字a、b、cが存在するかどうかを確認する必要があります。

したがって、入力が[10、2、8、5、6]の場合、出力はTrueになり、8 ^ 2 + 6 ^ 2 =64 + 36 =100 =10^2となります。

これを解決するには、次の手順に従います-

• tmp：=numsのすべての数値の降順の2乗のリスト
• tmp内の各インデックスiおよび対応する番号nについて、実行します
  • ベース：=n
  • 左：=i + 1、右：=tmp-1のサイズ
  • 左<=右、実行
    • t：=2つのリストtmp[left]とtmp[right]を結合します
    • tがbaseと同じ場合、
      • Trueを返す
    • それ以外の場合、t>ベースの場合、
      • 左：=左+ 1
    • それ以外の場合、
      • right：=right-1
• Falseを返す

理解を深めるために、次の実装を見てみましょう-

例

class Solution:
   def solve(self, nums):
      tmp = sorted([n*n for n in nums], reverse = True)
      for i, n in enumerate(tmp):
         base = n
         left = i+1; right = len(tmp)-1
         while left <= right:
            t = tmp[left]+tmp[right]
            if t == base:
               return True
            elif t > base:
               left += 1
            else:
               right -= 1
      return False
ob = Solution()
print(ob.solve([10, 2, 8, 5, 6]))

入力

[10, 2, 8, 5, 6]

出力

True