给定平面上 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]]
思路:对于任意一个点A来说,如果到3个点B、C、D的距离相等,那么会构成3*2个"boomerang",分别是A B C,AC B,A B D,AD B,A C D,AD C,同样对于任意一个点,如果到n个点距离相等,那么会构成n*(n-1)个"boomerang"。我们可以遍历数组中的每一个点,对于任意一个点,求出到它距离相等的点的个数,根据个数求出"boomerang"的数量,对数量进行累加,得到最终结果。注意一个语法:
hypot函数头文件math.h或cmath
hypot(a,b)的返回值为double类型,相当于sqrt(a*a+b*b)
class Solution {
public:
int numberOfBoomerangs(vector<vector<int>>& points) {
int re = 0;
for (int i = 0; i<points.size(); ++i){
unordered_map<double, int>m;
for (int j = 0; j<points.size(); ++j){
if (j == i) continue;
m[hypot(points[i][0] - points[j][0], points[i][1] - points[j][1])]++;
}
for (auto it = m.begin(); it != m.end(); ++it){
re += it->second*(it->second - 1);
}
}
return re;
}
};