力扣 447. 回旋镖的数量(中等)

题目

给定平面上 n 对 互不相同 的点 points ,其中 points[i] = [xi, yi] 。回旋镖 是由点 (i, j, k) 表示的元组 ,其中 i 和 j 之间的距离和 i 和 k 之间的距离相等(需要考虑元组的顺序)。
返回平面上所有回旋镖的数量。
示例 1:

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

示例 2:

输入:points = [[1,1],[2,2],[3,3]]
输出:2

示例 3:

输入:points = [[1,1]]
输出:0

提示:

  • n == points.length
  • 1 <= n <= 500
  • points[i].length == 2
  • − 1 0 4 < = x i , y i < = 1 0 4 -10^4 <= x_i, y_i <= 10^4 104<=xi,yi<=104

所有点都互不相同

题解

方法一:暴力遍历+哈希表
首先枚举每个中心点,遍历points,计算并统计所有点到到中心点的距离,将每个距离的出现次数记录在哈希表中,然后遍历哈希表。假设某个距离有m个点,那么要从这m个点中选出2个点和中心点构成回旋镖,由于题目说明要考虑点的顺序,所以方案数即为在 m 个元素中选出 2 个不同元素的排列数,即: A m 2 = m × ( m − 1 ) A^2_m=m×(m-1) Am2=m×(m1)

class Solution {
    public int numberOfBoomerangs(int[][] points) {
        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;
    }
}

你可能感兴趣的:(算法,线性代数)