[leetcode] 447. Number of Boomerangs

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/w5688414/article/details/89929538

Description

Given n points in the plane that are all pairwise distinct, a “boomerang” is a tuple of points (i, j, k) such that the distance between i and j equals the distance between i and k (the order of the tuple matters).

Find the number of boomerangs. You may assume that n will be at most 500 and coordinates of points are all in the range [-10000, 10000] (inclusive).

Example:

Input:

[[0,0],[1,0],[2,0]]

Output:

2

Explanation:

The two boomerangs are [[1,0],[0,0],[2,0]] and [[1,0],[2,0],[0,0]]

分析

题目的意思是:给你一些点,用这些点组成三元组(i,j,k),其中i到j的距离跟i到k的距离相等。

  • 如果我们有一个点a,还有两个点b和c,如果ab和ac之间的距离相等,那么就有两种排列方法abc和acb;如果有三个点b,c,d都分别和a之间的距离相等,那么有六种排列方法,abc, acb, acd, adc, abd, adb,类推:如果有n个点,那么排列方式为n*(n-1)
  • 那么我们问题就变成了遍历所有点,让每个点都做一次点a,然后遍历其他所有点,统计和a距离相等的点有多少个,然后分别带入n(n-1)计算结果并累加到res中,只有当n大于等于2时,res值才会真正增加。

代码

class Solution {
public:
    int numberOfBoomerangs(vector<pair<int, int>>& points) {
        int res=0;
        for(int i=0;i<points.size();i++){
            unordered_map<int,int> m;
            for(int j=0;j<points.size();j++){
                int a=points[j].first-points[i].first;
                int b=points[j].second-points[i].second;
                m[a*a+b*b]++;
            }
            for(auto t:m){
                res+=t.second*(t.second-1);
            }
        }
        return res;
    }
};

参考文献

[LeetCode] Number of Boomerangs 回旋镖的数量

猜你喜欢

转载自blog.csdn.net/w5688414/article/details/89929538