题目:
给定平面上 n 对 互不相同 的点 points ,其中 points[i] = [xi, yi] 。回旋镖 是由点 (i, j, k) 表示的元组 ,其中 i 和 j 之间的距离和 i 和 k 之间的距离相等(需要考虑元组的顺序)。
返回平面上所有回旋镖的数量。
思路:
对points中的每个点,统计其与points中任意点的距离,并用字典进行记录每种距离出现的次数。并用A(m,2)来统计所有可能的情况。
解答:
class Solution:
def numberOfBoomerangs(self, points: List[List[int]]) -> int:
ans = 0
for p in points:
cnt = defaultdict(int)
for q in points:
dis = (p[0] - q[0]) **2 + (p[1] - q[1]) **2
cnt[dis] += 1
for m in cnt.values():
ans += m * (m - 1)
#print(cnt)
return ans