447. 回旋镖的数量

447. 回旋镖的数量

难度中等162

给定平面上 n 对 互不相同 的点 points ，其中 points[i] = [xi, yi] 。回旋镖 是由点 (i, j, k) 表示的元组 ，其中 i 和 j 之间的距离和 i 和 k 之间的距离相等（需要考虑元组的顺序）。

返回平面上所有回旋镖的数量。

示例 1：

输入：points = [[0,0],[1,0],[2,0]]
输出：2
解释：两个回旋镖为 [[1,0],[0,0],[2,0]] 和 [[1,0],[2,0],[0,0]]

示例 2：

输入：points = [[1,1],[2,2],[3,3]]
输出：2

示例 3：

输入：points = [[1,1]]
输出：0

提示：

• n == points.length
• 1 <= n <= 500
• points[i].length == 2
• -104 <= xi, yi <= 104
• 所有点都 互不相同

思路：由于点的数量很有限(<=500)，因此我们可以直接枚举i，然后遍历整个数组，以距离为key统计与i距离为dis的点的数量，记为num_count[dis]，那么当我们需要求解元组(i,j,k)的时候，假设当前讨论的距离是dis，那么我们只需要从num_count[dis]对应的集合里取出两个点i,j，就可以构成(i,j,k)与(i,k,j)，因此每一个num=num_couns[dis]对i的答案贡献为 num * (num - 1) / 2 * 2 [*2是因为j与k可以交换位置]。

[注]：num_count统计结束后需要恢复初始化状态。

class Solution {
public:
    int numberOfBoomerangs(vector<vector<int>>& points) {
        int i, j, n = points.size(), dis, ans = 0, num;
        unordered_map<int, int> num_count;
        for(i = 0; i < n; ++ i){
            num_count.clear();
            for(j = 0; j < n; ++ j){
                if(i == j) continue;
                dis = (points[i][0] - points[j][0]) * (points[i][0] - points[j][0]) + (points[i][1] - points[j][1]) * (points[i][1] - points[j][1]);
                num_count[dis] += 1;
            }
            for(unordered_map<int, int>::iterator iter = num_count.begin(); iter != num_count.end(); ++ iter) {
                num = iter->second;
                ans = ans + num * (num - 1);
            }
        }
        return ans;
    }
};