[LeetCode] 每日一题 Day 21：447. 回旋镖的数量

题目

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

找到所有回旋镖的数量。你可以假设 n 最大为 500，所有点的坐标在闭区间 [-10000, 10000] 中。

示例:

输入:	[[0,0],[1,0],[2,0]]
输出:	2
解释:
两个回旋镖为 [[1,0],[0,0],[2,0]] 和 [[1,0],[2,0],[0,0]]

解决

思路
用哈希表存储一下以每个位置为起点时，距离其他位置的距离，然后遍历哈希表用组合数求一下数量，假如相同距离的点有n个，那么数量就是$A_n^2$

代码

class Solution {
    public int numberOfBoomerangs(int[][] points) {
        int res=0;
        for(int i=0;i<points.length;i++){
            Map<Integer, Integer> map = new HashMap<Integer, Integer>();
            for(int j=0;j<points.length;j++){
                map.put(dis(points[i],points[j]),map.getOrDefault(dis(points[i],points[j]),0)+1);
            }
            for(Map.Entry<Integer, Integer> entry : map.entrySet()){
                res+=entry.getValue()*(entry.getValue()-1);
            }
        }
        return res;
    }
    public int dis(int[] x, int[] y){
        
        return (int)(Math.pow((x[0]-y[0]),2)+Math.pow((x[1]-y[1]),2));
    }
}