LeetCode刷题笔记（Number of Boomerangs）

北京这几天wumai有点严重，lianghui期间出现这种事，某某部门人事变动可能会有点大吧。言归正传，刚刚又刷了一道题，感觉收获不小，下面和大家来分享一下吧！

题目如下：

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个点且这些点都是成对不同的，“boomerang”是由三个点（i，j，k）构成的元组，该元组满足使得i和j之间的距离等于i和k之间的距离，另外元组元素的顺序不同可能组成不同的“boomerang”。最后请返回boomerang的数目。

注：①点的数目最多为500个；②点的坐标值在[-10000, 10000] 之间。

解答如下：

方法一（unordered_map查找法）

先用一层for循环依次固定每个点，然后为这每个点创建一个unordered_map，其键为其他点到该点的距离，其值为到该点距离相等点的数目，最后用迭代器来遍历unordered_map，将键频次大于2的值进行计算并存放在result中，返回result。

注：①这里计算距离时，不要开根号，否则浮点型的数不利于unordered_map的查找，其实开根号与不开根号并不影响最终的比较结果。②迭代器的生成，以及元素的访问要注意。③ pair<int, int>的赋值以及访问也要注意。

class Solution{
public:
    int numberOfBoomerangs(vector< pair<int, int>>& points){
        int len = points.size();
        int result = 0;
        for (int i = 0; i < len ; i++){
            unordered_map <int, int> record;
            for (int j = 0; j < len ; j++) {
                if(j != i) 
                //记录到points[i]距离相等点的数目；points[i].first访问pair型变量第一个值
                record[(points[i].first - points[j].first) * (points[i].first - points[j].first) +
                (points[i].second - points[j].second) * (points[i].second - points[j].second)]++;
            }
            //用迭代器来遍历record；iter -> second访问迭代器元素第二个值   
            for (unordered_map <int, int> :: iterator iter = record.begin(); iter != record.end(); iter++) {
                if(iter -> second >= 2)
                {result += (iter -> second) * (iter -> second - 1);
                    cout << iter -> second << " " << iter -> second -1<< endl;}
            }
        }
        return  result;
    }
};

 

提交后的结果如下：

 

方法二（优化方法一） 

①将求两点距离的公式，单独定义成一个私有函数。

②去掉int len = points.size();，循环中直接用points.size()；

③去掉if(iter -> second >= 2)判断一样可行，但去掉后提交时会出现Output Limit Exceeded。

 

class Solution{
public:
    int numberOfBoomerangs(vector< pair<int, int>>& points){
        int result = 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[distance(points[i], points[j])]++;
            }
            for (unordered_map <int, int> :: iterator iter = record.begin(); iter != record.end(); iter++) {
                if(iter -> second >= 2)
                {result += (iter -> second) * (iter -> second - 1);
                    cout << iter -> second << " " << iter -> second -1<< endl;}
            }
        }
        return  result;
    }

private:  //定义的距离函数
    int distance(const pair<int,int>& x, const pair<int,int> y){
        return (x.first - y.first) * (x.first - y.first) +
               (x.second - y.second) * (x.second - y.second);
    };
};

 

提交后的结果如下：

日积月累，与君共进，增增小结，未完待续。