力扣447. 回旋镖的数量

class Solution {
    public int numberOfBoomerangs(int[][] points) {
        int ans = 0;
        for(int[] p :points){
            Map dict = new HashMap();
            for(int[] q:points){
                int len = (q[0]-p[0])*(q[0]-p[0])+(q[1]-p[1])*(q[1]-p[1]);
                dict.put(len,dict.getOrDefault(len,0)+1);
            }
            for (Integer m : dict.values()){
                ans += m*(m-1);
            }
        }
        return ans;
    }
}

方法一:枚举 + 哈希表
题目所描述的回旋镖可以视作一个 \texttt{V}V 型的折线。我们可以枚举每个 \textit{points}[i]points[i],将其当作 \texttt{V}V 型的拐点。设 \textit{points}points 中有 mm 个点到 \textit{points}[i]points[i] 的距离均相等,我们需要从这 mm 点中选出 22 个点当作回旋镖的 22 个端点,由于题目要求考虑元组的顺序,因此方案数即为在 mm 个元素中选出 22 个不同元素的排列数,即:

A_m^2 = m\cdot(m-1)

m
2

 =m⋅(m−1)

据此,我们可以遍历 \textit{points}points,计算并统计所有点到 \textit{points}[i]points[i] 的距离,将每个距离的出现次数记录在哈希表中,然后遍历哈希表,并用上述公式计算并累加回旋镖的个数。

在代码实现时,我们可以直接保存距离的平方,避免复杂的开方运算。

作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/number-of-boomerangs/solution/hui-xuan-biao-de-shu-liang-by-leetcode-s-lft5/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

你可能感兴趣的:(力扣,leetcode,算法,python)