首页 > 其他 > 详细

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

时间:2019-03-25 10:44:07      阅读:125      评论:0      收藏:0      [点我收藏+]

题目描述:

给定平面上 n 对不同的点,“回旋镖” 是由点表示的元组 (i, j, k) ,其中 ij 之间的距离和 ik 之间的距离相等(需要考虑元组的顺序)。

找到所有回旋镖的数量。你可以假设 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;
    }
};

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

原文:https://www.cnblogs.com/zhanzq/p/10592317.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!