447. 回旋镖的数量
给定平面上 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]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-boomerangs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
哈希表
题目可以这样理解:对于一个点,寻找是否有两个及两个以上的点到它的距离相同。按照题意应该遍历两次,第一次确定点,第二次于其他点的计算距离和找到距离相同的点。
class Solution:
def numberOfBoomerangs(self, points: List[List[int]]) -> int:
result = 0
for s in points:
dic = {}
for t in points:
dis = (s[0]-t[0])**2+(s[1]-t[1])**2
if dis in dic:
dic[dis] += 1
else:
dic[dis] = 1
for k in dic:
result += dic[k]*(dic[k]-1)
return result