来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-boomerangs
题目描述:
给定平面上 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
题目分析:
- 点i到点j的距离等于点i到点k的距离,则是一个回旋镖
- 需要考虑元组顺序.
思路分析:
计算n对点,两两之间的距离,并记录具有相同距离的点的数量。使用排列组合计算每个距离的回旋镖数量再进行累加。
代码实现:
class Solution {
public int result = 0;
public int numberOfBoomerangs(int[][] points) {
int len = points.length;
for (int i = 0; i < len; i++) {
Map<Integer, Integer> map = new HashMap();
int left_src_num = points[i][0];
int right_src_num = points[i][1];
for (int j = 0; j < len; j++) {
if (i == j) continue; // 跳过同一个节点
int left_num = points[j][0];
int right_num = points[j][1];
int key = (left_src_num - left_num) * (left_src_num - left_num)
+ (right_src_num - right_num) * (right_src_num - right_num);
map.put(key, map.getOrDefault(key, 0) + 1);
}
for (Map.Entry<Integer, Integer> entrySet : map.entrySet()) {
int value = entrySet.getValue();
result += value * (value - 1); // 需要考虑元组顺序,全排列
}
}
return result;
}
}