1. 问题描述:
给定平面上 n 对互不相同的点 points ,其中 points[i] = [xi, yi] 。回旋镖是由点 (i, j, k) 表示的元组 ,其中 i 和 j 之间的距离和 i 和 k 之间的距离相等(需要考虑元组的顺序)。返回平面上所有回旋镖的数量。
示例 1:
输入:points = [[0,0],[1,0],[2,0]]
输出:2
解释:两个回旋镖为 [[1,0],[0,0],[2,0]] 和 [[1,0],[2,0],[0,0]]
示例 2:
输入:points = [[1,1],[2,2],[3,3]]
输出:2
示例 3:
输入:points = [[1,1]]
输出:0
提示:
n == points.length
1 <= n <= 500
points[i].length == 2
-10 ^ 4 <= xi, yi <= 10 ^ 4
所有点都互不相同
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-boomerangs
2. 思路分析:
分析题目可以知道最简单的做法是使用三层循环暴力枚举,但是有可能会超时,我们还可以对其优化一下,我们其实可以使用枚举每一类点,使用哈希表将距离相等的点存储在一起,也即对距离相等的点进行计数,为了避免开根号导致的精度问题我们可以在哈希表中存储距离的平方,这样我们可以使用两层循环解决,第一层循环用来枚举三元组中的第一个点,第二层循环用来枚举其余的两个点,然后枚举字典计算以当前的第i个点为第一个点到达j, k两个点距离相等的组合数目。
3. 代码如下:
from typing import List
import collections
class Solution:
def numberOfBoomerangs(self, p: List[List[int]]) -> int:
res = 0
for i in range(len(p)):
dic = collections.defaultdict(int)
for j in range(len(p)):
if i != j:
dx = p[i][0] - p[j][0]
dy = p[i][1] - p[j][1]
# 为了避免精度的问题所以存储距离的平方
dic[dx ** 2 + dy ** 2] += 1
# 枚举答案的数目
for k, v in dic.items():
# 需要考虑其余两个点的顺序所以不用除以2
res += v * (v - 1)
return res