Pythonのブーメランの数

したがって、入力が[[0,0]、[1,0]、[2,0]]の場合、2つのブーメランは[[1,0]、[0,0]であるため、出力は2になります。 、[2,0]]および[[1,0]、[2,0]、[0,0]]。

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

• counter_of_boomerangs：=0

• ポイント配列のpoint_1ごとに、次のようにします

  • x1、y1 =point_1

  • distance_count_dict

    というマップを定義します

  • ポイント配列のpoint_2ごとに、次のようにします

    • x2、y2 =point_2

    • diff_x：=x2-x1

    • diff_y：=y2-y1

    • dist：=diff_x ^ 2 + diff_y ^ 2

    • distance_count_dict [dist]：=distance_count_dict [dist] + 1

  • distance_count_dictの各dに対して-

    • n：=distance_count_dict [d]

    • counter_of_boomerangs：=counter_of_boomerangs + n *（n-1）

• counter_of_boomerangsを返す

例

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

from collections import defaultdict
class Solution:
   def numberOfBoomerangs(self, points):
      counter_of_boomerangs = 0
      for point_1 in points:
         x1, y1 = point_1
         distance_count_dict = defaultdict( int )
         for point_2 in points:
            x2, y2 = point_2
            diff_x = x2-x1
            diff_y = y2-y1
            dist = diff_x ** 2 + diff_y ** 2
            distance_count_dict[ dist ] += 1
         for d in distance_count_dict:
            n = distance_count_dict[d]
            counter_of_boomerangs += n * (n-1)
      return counter_of_boomerangs

ob = Solution()
print(ob.numberOfBoomerangs([[0,0],[1,0],[2,0]]))

入力

[[0,0],[1,0],[2,0]]

出力

0