题目描述:(来源于力扣447)
给定平面上 n 对不同的点,“回旋镖” 是由点表示的元组 (i, j, k) ,其中 i 和 j 之间的距离和 i 和 k 之间的距离相等(需要考虑元组的顺序)。
找到所有回旋镖的数量。你可以假设 n 最大为 500,所有点的坐标在闭区间 [-10000, 10000] 中。
示例:
题目中的表述有点绕,其实就是两边节点到中间节点的距离相等。
这里的n最多为500,如果我们使用暴力解法,时间复杂度是 O ( n 3 ) O(n^3) O(n3),显然会超时。
观察到 i 是一个“枢纽”,对于每个点 i ,遍历 j 、k 点到 i的距离,这样时间复杂度就是 O ( n 2 ) O(n^2) O(n2)的。
因此,转化成求距离 i 有相同距离的点有多少个。
注意题中是要求元组的顺序的,所以下面计算点到i的距离就需要考虑点的排列。
注意:
1)计算两点之间的距离时,为了避免浮点数,这里采用整数,计算平方差;
2)注意距离返回值是否溢出,这道题不会溢出;
class Solution {
public:
int numberOfBoomerangs(vector<vector<int>>& points) {
int res=0;
int dis=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){
// 计算距离时不进行开根运算, 以保证精度
dis=pow(points[i][0]-points[j][0],2)+pow(points[i][1]-points[j][1],2);
record[dis]++;
}
for(unordered_map<int, int>::iterator iter = record.begin() ; iter != record.end() ; iter ++)
{
if((*iter).second>=2)
res+=((*iter).second)*((*iter).second-1);
}
}
return res;
}
};