给定平面上 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
方法1:创建一个map<int, int> key值代表距第i个点的距离,value代表个数。
(1)从第一个点开始遍历,算出其他所有点的距离,并统计个数,一个点算完之后,开始统计个数。
(2)同(1)开始第二个点,循环结束即可。
class Solution {
public:
int numberOfBoomerangs(vector<vector<int>>& points) {
int res = 0;
for(int i= 0; i < points.size();i++){
unordered_map<int,int> record;
for(int j = 0; j< points.size();j++)
if(j !=i)
record[ dis(points[i],points[j]) ] ++; //
for(unordered_map<int,int>::iterator iter = record.begin(); iter !=record.end(); iter++)
res = res + iter->second *(iter->second-1);
}
return res;
}
private:
int dis(vector<int> pa, vector<int> pb){
return (pa[0]-pb[0])*(pa[0]-pb[0]) + (pa[1]-pb[1])*(pa[1]-pb[1]);
}
};