【LeetCode 204】关关的刷题日记40 Number of Boomerangs- 专知

【LeetCode 204】关关的刷题日记40 Number of Boomerangs

点击上方“专知”，关注获取专业AI知识”

关关的刷题日记40 – Leetcode 447. Number of Boomerangs

题目

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.

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

题目的意思是一个平面上有n个点，一个"boomerang"由三个点(i, j, k)构成，i和j之间的距离等于i和k之间的距离，让我们求这n个点构成的"boomerang"的数量。

思路

思路：对于任意一个点P来说，如果到3个点B、C、D的距离相等，那么会构成A32 个"boomerang"，分别是 A B C，A C B，A B D，A D B，A C D，A D C，同样对于任意一个点，如果到n个点距离相等，那么会构成An2 个"boomerang"。我们可以遍历数组中的每一个点，对于任意一个点，求出到它距离相等的点的个数，根据个数求出"boomerang"的数量，对数量进行累加，得到最终结果。

class Solution {public:    int numberOfBoomerangs(vector<pair<int, int>>& points) {        int n=points.size(), re=0, dis=0;        for(int i=0;i<n;i++)        {            map<int,int>m;            for(int j=0; j<n; j++)            {                dis=(points[i].first-points[j].first)*(points[i].first-points[j].first)+(points[i].second-points[j].second)*(points[i].second-points[j].second);                m[dis]++;            }            for(auto x:m)            {                if(x.second>=2)                    re+=(x.second-1)*x.second;            }        }        return re;    }};