问题描述
给定平面上 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]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-boomerangs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
执行结果
代码描述
思路:以每一个点为中点,然后求其他各点到这个点的距离,距离相等的进行累加。假设到某点A的距离为2的点,pairs[2] = 3,则此时计算count = = 3*2 = 6. 然后,又增加了一个点,到A的距离为2,pairs[2] = 4, 每增加一个点,count += 2 * (pairs[2] - 1). 因为分方向,所以需要2倍的前一次的pairs[2]的数量。
class Solution {
public:
int numberOfBoomerangs(vector<vector<int>>& points) {
if(points.size() < 3) return 0;
if(points[0].size() < 2) return 0;
int count = 0;
map<int, int> pairs;
for(int i = 0; i < points.size(); ++i)
{
pairs.clear();
for(int j = 0; j < points.size(); ++j)
{
if(i != j)
{
int temp1 = pow((points[i][0] - points[j][0]), 2);
int temp2 = pow((points[i][1] - points[j][1]), 2);
int a = temp1 + temp2;
pairs[a] += 1;
if(pairs[a] > 1)
count += 2*(pairs[a]-1);
}
}
}
return count;
}
};