回旋镖的数量

来源:力扣(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

题目分析:
  1. 点i到点j的距离等于点i到点k的距离,则是一个回旋镖
  2. 需要考虑元组顺序.
思路分析:

计算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;
    }
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容