Python3解leetcode Number of Boomerangs

问题描述：

Given n points in the plane that are all pairwise distinct, a "boomerang" is a tuple of points (i, j, k) such that the distance between i and jequals the distance between i and k (the order of the tuple matters).

Find the number of boomerangs. You may assume that n will be at most 500 and coordinates of points are all in the range [-10000, 10000](inclusive).

Example:

Input:
[[0,0],[1,0],[2,0]]

Output:
2

Explanation:
The two boomerangs are [[1,0],[0,0],[2,0]] and [[1,0],[2,0],[0,0]]

思路：主要学习np.unique()-去重并排序，同时根据return_counts参数可以确定各个元素的个数,squareform()-将点之间距离在简洁和冗余模式相互转化,pdist()-求点之间距离，等函数的应用

代码：

 import numpy as np
 from numpy import *
 from scipy.spatial.distance import pdist, squareform

 class Solution:
     def numberOfBoomerangs(self, points: List[List[int]]) -> int:
         a = squareform(pdist(np.array(points)))

         result = 0
         for i in a:#遍历每一行
             count = np.unique(i,return_counts=True)[1]
             result += sum(count*(count - 1))

         return result