LeetCode-Number of Boomerangs

最新推荐文章于 2020-11-03 20:31:43 发布

原创最新推荐文章于 2020-11-03 20:31:43 发布 · 133 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #Hash Table

LeetCode 专栏收录该内容

306 篇文章

订阅专栏

本文介绍了一种算法，用于从给定的二维平面上的多个点中找出所有可能构成‘回旋镖’的点集，即满足两点到中心点等距的三元组。通过遍历每个点并使用散列表来跟踪距离相同的点的数量，该算法能有效地计算出所有符合条件的‘回旋镖’数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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)$ ，使得满足 $d i s (i, j) = = d i s (i, k)$ ；

解法：我们需要考虑每一个点，计算是否有另外的两个点满足到当前点的距离相同；

遍历所有的点 $i$ ；
遍历除 $i$ 外的所有点 $j$ ，利用散列表 $t a b l e$ 记录下此时点 $j$ 到点 $i$ 的距离，并且对于每个点 $j$ ，在散列表中查找是否有其他的点到 $i$ 的距离相同；则此时可能的坐标点组合为 $2 * t a b l e (d i s (i, j))$ ，每包含一个具有相同距离的点，其组合数增加2（即 $(i, j, k)$ 和 $(i, k, j)$ ）；

Java

class Solution {
    public int numberOfBoomerangs(int[][] points) {
        int result = 0;
        for (int[] i : points) {
            Map<Integer, Integer> table = new HashMap<>();
            for (int[] j : points) {
                if (i == j) continue;
                int dis = (i[0] - j[0]) * (i[0] - j[0]) + 
                    (i[1] - j[1]) * (i[1] - j[1]);
                int preDis = table.getOrDefault(dis, 0);
                result += 2 * preDis;
                table.put(dis, preDis + 1);
            }
        }
        return result;
    }
}