447. 回旋镖的数量(JavaScript)

给定平面上 n 对不同的点,“回旋镖” 是由点表示的元组 (i, j, k) ,其中 i 和 j 之间的距离和 i 和 k 之间的距离相等(需要考虑元组的顺序)。

找到所有回旋镖的数量。你可以假设 n 最大为 500,所有点的坐标在闭区间 [-10000, 10000] 中。

示例:

输入:
[[0,0],[1,0],[2,0]]

输出:
2

解释:
两个回旋镖为 [[1,0],[0,0],[2,0]][[1,0],[2,0],[0,0]]

思路:

对于每个点,计算其他各点中,所有的距离与该距离的点数。使用map存储。

js中的map用法:

let map = new Map()    //初始化
map.set(a, b)          //设置a的值为b
map.get(a)             //b

每一次循环,都计算以该点为轴心,共能有几个回旋镖。有n个点距离此点的距离相同,那么可以组成n*(n-1)个回旋镖。

/**
 * @param {number[][]} points
 * @return {number}
 */
const numberOfBoomerangs = function(points) {
  let result = 0,
      length = points.length;
  const getDistance = (point1, point2) => {
    return Math.pow(point1[0] - point2[0], 2) + Math.pow(point1[1] - point2[1], 2);
  }
  for (let i = 0; i < length; i++) {
    let map = new Map();
    for (let j = 0; j < length; j++) {
      if (i === j) continue;
      let distance = getDistance(points[i], points[j])
      map.set(distance, (map.get(distance) || 0) + 1)
    }
    map.forEach( (value) => { result += value * (value - 1) })
  }
  return result;
};

 

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务