Leetcode算法学习日志-447 Number of Boomerangs

Leetcode 447 Number of Boomerangs

题目原文

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]]

题意分析

给定n个点及它们的坐标，求出"boomerangs"的个数，（i，j，k）一组点，如果i到j的距离和i到k的距离一样，则这样一组点称为"boomerangs"，j和k的顺序可以调换得到另一个“boomerangs”。

解法分析

要从n个点中选取3个点，不同的选取方法（考虑顺序）个数为O(n^3)，因此算法能达到O(n^2)即可。对于每一个点，遍历其他点，将距离作为关键字，有相应距离的点的个数为关联值，建立一个unordered_map，对于本题的情况，unordered_map采用直接寻址表，查询、插入、搜索操作的复杂度都是O(1)，就算是桶内有多个元素的采用链接法的散列表，如果哈希函数选得均匀，字典操作复杂度也为O(1)。建好表后，用范围for语句遍历表，对于大于一的关联值求增加的“boomerangs”数。C++代码如下：

class Solution {
public:
int numberOfBoomerangs(vector<pair<int, int>>& points) {
int res=0;
unordered_map<int,int> distance;//first is the distance, second is the number of points that the same distance with point i
int i,j,key;
for(i=0;i<points.size();i++){
distance.clear();
for(j=0;j<points.size();j++)
{
if(j==i)
continue;
key=(points[i].first-points[j].first)*(points[i].first-points[j].first)+(points[i].second-points[j].second)*(points[i].second-points[j].second);
distance[key]++;
}
for(auto p:distance){
if(p.second>1)
res+=p.second*(p.second-1);
}
}
return res;
}
};由于map的操作复杂度都是O(1)，因此算法复杂度为O(n^2)。