给定平面上 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
-10^4 <= xi, yi <= 10^4
回旋镖 是由点 (i, j, k)
表示的元组 ,其中 i
和 j
之间的距离和 i
和 k
之间的欧式距离相等(需要考虑元组的顺序)。
欧式距离:
即我们需要以i为顶点,统计距离(dis)相等的顶点个数cnt,在从中选取2个(j,k),组成回旋镖,而且需要考虑顺序,即排列。
那么对于顶点i,回旋镖的数量为=从cnt中选2个的排列=A(cnt,2)=cnt*(cnt-1)
根据3的描述,数据结构可以使用map,key是dis,val是cnt
HashMap<Integer, Integer> disCount = new HashMap<>();
对于每个点都做顶点,循环计算回旋镖数量,再累计求和即可。
会有一种情况,p1点和p2点是同一个点是,dis=0对应的以p1为顶点的点的个数是1
因为所有点都是互相不同的,所以双层for循环,dis=0,只会有一个点,不能构成回旋镖,可以不用理会,相当于ans+=0。
class Solution {
public int numberOfBoomerangs(int[][] points) {
int ans = 0;
//key是距离,value是距离相等的顶点个数
HashMap<Integer, Integer> disCount = new HashMap<>();
for (int[] point1 : points) {
disCount.clear();
for (int[] point2 : points) {
int dis = (point1[0] - point2[0]) * (point1[0] - point2[0]) + (point1[1] - point2[1]) * (point1[1] - point2[1]);
disCount.put(dis, disCount.getOrDefault(dis, 0) + 1);
}
for (Integer cnt : disCount.values()) {
ans += cnt * (cnt - 1);
}
}
return ans;
}
}
qq群: