LeetCode—— 447 回旋镖的数量

问题描述

给定平面上 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]]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-boomerangs
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：以每一个点为中点，然后求其他各点到这个点的距离，距离相等的进行累加。假设到某点A的距离为2的点，pairs[2] = 3，则此时计算count =  = 3*2 = 6. 然后，又增加了一个点，到A的距离为2，pairs[2] = 4, 每增加一个点，count += 2 * (pairs[2] - 1). 因为分方向，所以需要2倍的前一次的pairs[2]的数量。

class Solution {
public:
    int numberOfBoomerangs(vector<vector<int>>& points) {
        if(points.size() < 3) return 0;
        if(points[0].size() < 2)    return 0;
        int count = 0;
        map<int, int> pairs;
        for(int i = 0; i < points.size(); ++i)
        {
            pairs.clear();
            for(int j = 0; j < points.size(); ++j)
            {
                if(i != j)
                {
                    int temp1 = pow((points[i][0] - points[j][0]), 2);
                    int temp2 = pow((points[i][1] - points[j][1]), 2);
                    int a = temp1 + temp2;
                    pairs[a] += 1;
                    if(pairs[a] > 1)
                        count += 2*(pairs[a]-1);
                }
            }
        }
        return count;
    }
};