原题:
Given n points in the plane that are all pairwise distinct, a "boomerang" is a tuple of points (i, j, k)
such that the distance between i
and j
equals the distance between i
and k
(the order of the tuple matters).
Find the number of boomerangs. You may assume that n will be at most 500 and coordinates of points are all in the range [-10000, 10000] (inclusive).
Example:
Input: [[0,0],[1,0],[2,0]] Output: 2 Explanation: The two boomerangs are [[1,0],[0,0],[2,0]] and [[1,0],[2,0],[0,0]]
题意:要求计算所有满足到点i的距离相等的有序元组[i, j, k],其中i、j、k代表坐标点。
解决方案:对每个点i,计算其他所有点到点i的距离,并用一个map容器m存储每个距离所对应的点的个数,对于每个距离d,每新增一个点j使得i与j之间的距离等于d,只需将ans加上2*m[d](由于j、k有序),并将m[d]++即可。
代码实现如下:
class Solution {
public:
int numberOfBoomerangs(vector<pair<int, int> >& points) {
int n = points.size(), ans = 0;
for(int i=0; i<n; i++)
{
map<int, int> m;
for(int j=0; j<n; j++)
{
int dx = points[i].first-points[j].first, dy = points[i].second-points[j].second;
ans += m[dx*dx + dy*dy] ++;
}
}
return 2*ans;
}
};