leetcode 447. 回旋镖的数量(Number of Boomerangs)

题目描述：

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

---

解法：

# define PR pair<long long, long long>

class Solution {
public:
    long long getDist(const PR& pr1, const PR& pr2){
        long long res = 0;
        res += (pr1.first - pr2.first)*(pr1.first - pr2.first) + (pr1.second - pr2.second)*(pr1.second - pr2.second);
        return res;
    }
    
    int numberOfBoomerangs(vector<pair<int, int>>& points) {
        int n = points.size();
        vector<vector<long long>> dist(n, vector<long long>(n, 0));
        for(int i = 0; i < n; i++){
            for(int j = i+1; j < n; j++){
                dist[i][j] = getDist(points[i], points[j]);
                dist[j][i] = dist[i][j];
            }
        }
        int res = 0;
        for(int i = 0; i < n; i++){
            unordered_map<long long, int> mp;
            for(int j = 0; j < n; j++){
                if(mp.find(dist[i][j]) == mp.end()){
                    mp[dist[i][j]] = 1;
                }else{
                    mp[dist[i][j]]++;
                }
            }
            for(auto it : mp){
                res += it.second*(it.second-1);
            }
        }
        return res;
    }
};