LeetCode——447. 回旋镖的数量

玄昌盛不会编程2024-01-09 14:12

大佬，牛！！！

题目：给你一个n*2的数组，表示n个点。然后让你从中选择三个点i,j,k，使得i到j和i到k的欧氏距离相等。问一共有多少种情况。需要注意的是，假设i,j,k是满足条件的，那么i,k,j其实也是满足条件的，并且这是两种情况。
我的思路：我的思路就是先来两个for循环，然后确定前面两个点，然后根据几何法找第三个点。事先我已经把点都放在一个set集合了，set是string的，内容是点的坐标，然后用逗号拼接。这里说的几何法就是找已知两个点的x的差距（xGap）和y的差距（yGap），然后让i点的x加减xGap和Ygap、i点的y加减xGap和Ygap，最后组合一下。注意重叠的情况就好了，但是目前测下来不太对。
大佬的思路：大佬的思路就是for循环点，然后构建点i到其他点的map集合，key表示距离，value表示数量。然后遍历map，如果map的value是m，则其实有A(m，2)=m*(m-1)种情况。
技巧：哈希

java代码

java 复制代码

class Solution {
    public int numberOfBoomerangs(int[][] points) {
        if (points.length < 3) {
            return 0;
        }
        int ans = 0;
        for (int[] p : points) {
            Map<Integer, Integer> cnt = new HashMap<Integer, Integer>();
            for (int[] q : points) {
                int dis = (p[0] - q[0]) * (p[0] - q[0]) + (p[1] - q[1]) * (p[1] - q[1]);
                cnt.put(dis, cnt.getOrDefault(dis, 0) + 1);
            }
            for (Map.Entry<Integer, Integer> entry : cnt.entrySet()) {
                int m = entry.getValue();
                ans += m * (m - 1);
            }
        }
        return ans;
    }
}

总结：其实解题的关键在于这个map的构建，感觉这个题跟leetcode的第一题有异曲同工之妙。