leetcode 回旋镖的数量

给定平面上 n 个不同的点，“回旋镖” 是由点表示的元组 (i, j, k) ，其中 i 和 j 之间的距离和 i 和 k 之间的距离相等（需要考虑顺序）。找到所有回旋镖的数量。可以假设 n 最大为 500，所有点的坐标在闭区间 [-10000, 10000] 中。

示例:

输入:
[[0,0],[1,0],[2,0]]

输出:
2

解释:
两个回旋镖为 [[1,0],[0,0],[2,0]] 和 [[1,0],[2,0],[0,0]]
链接：https://leetcode-cn.com/problems/number-of-boomerangs
 

class Solution(object):
    def numberOfBoomerangs(self, points):
        """
        :type points: List[List[int]]
        :rtype: int
        """
        num = 0
        for i in points:
            dist2 = dict()
            for j in points:
                if i != j:
                    key = (i[0]-j[0])**2+(i[1]-j[1])**2
                    dist2[key] = dist2.get(key, 0) + 1
            for times in dist2.values():
                if times>1:
                    num += times*(times-1)
        return num