447 Number of Boomerangs 回旋镖的数量

Posted 2020-10-31 lina2014

给定平面上 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]]
详见：https://leetcode.com/problems/number-of-boomerangs/description/

C++：

class Solution {
public:
    int numberOfBoomerangs(vector<pair<int, int>>& points)
    {
        int res = 0;
        for (int i = 0; i < points.size(); ++i)
        {
            unordered_map<int, int> m;
            for (int j = 0; j < points.size(); ++j)
            {
                int a = points[i].first - points[j].first;
                int b = points[i].second - points[j].second;
                ++m[a * a + b * b];
            }
            for (auto it = m.begin(); it != m.end(); ++it) 
            {
                res += it->second * (it->second - 1);
            }
        }
        return res;
    }
};

 参考：https://leetcode.com/problems/number-of-boomerangs/description/