leetcode中等447回旋镖的数量

Posted 2021-10-02 qq_40707462

思路：遍历+hash
对于一个点p，假设3个点到它距离相同，那一共能产生3 * 2组回旋镖;m个点到它距离相同，那一共能产生m * (m-1)组回旋镖。
用hash记录每个距离出现的次数，累加即可。

class Solution:
    def numberOfBoomerangs(self, points: List[List[int]]) -> int:
        ans = 0
        for p in points:
            hash = {}
            for q in points:
                dis = (p[0] - q[0]) * (p[0] - q[0]) + (p[1] - q[1]) * (p[1] - q[1])
                hash[dis]=hash.get(dis,0)+1
            for m in hash.values():
                ans += m * (m - 1)
        return ans

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