leetcode447. 回旋镖的数量

给定平面上 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]]

n个相同的距离有n*(n-1)个,n+1个相同距离有n*(n+1)个,n增加1,结果增加2*n:

class Solution:
    def numberOfBoomerangs(self, points: List[List[int]]) -> int:
        dist_dict = {}  # 相同距离出现的次数
        res = 0
        for i in range(len(points)):
            for j in range(len(points)):
                if i != j:
                    dist = (points[i][0]-points[j][0])**2 + (points[i][1]-points[j][1])**2
                    if dist not in dist_dict:
                        dist_dict[dist] = 1
                    else:
                        res += 2*dist_dict[dist]  # 如果数量增加1,结果增加2n
                        dist_dict[dist] += 1
            dist_dict = {}
        return res
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章