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

 

解题思路

class Solution {
public:
    int numberOfBoomerangs(vector<vector<int>>& points) {
        unordered_map<long long,int> mp;
        int ans = 0,len = points.size();
        for(int i=0;i<len;i++){
            mp.clear();
            int x0 = points[i][0],y0 = points[i][1];
            for(int j=0;j<len;j++){
                if(i == j) continue;
                int x1 = points[j][0],y1 = points[j][1];
                long long dis = (x1-x0)*(x1-x0) + (y1-y0)*(y1-y0);
                mp[dis]++;
            }
            for(auto it=mp.begin();it!=mp.end();it++){
                int cnt = it->second;
                ans += cnt*(cnt-1);
            }
        }
        return ans;
    }
};