题目
给定平面上 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×(m−1)。
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;
}
}