题目:给定平面上 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]]
思路:没有什么好的思路,暴力枚举:遍历每个点,计算其他的点到这个点的距离,然后再找距离相等的点,
代码:
class Solution(object):
def numberOfBoomerangs(self, points):
"""
:type points: List[List[int]]
:rtype: int
"""
dist = {}
result = 0
for i in range(len(points)):
for j in range(i, len(points)):
distij = (points[i][0] - points[j][0])**2 + (points[i][1] - points[j][1])**2
dist[i] = dist.get(i, []) + [distij]
if not i == j:
dist[j] = dist.get(j, []) + [distij]
for i in range(len(points)):
dst = sorted(dist[i])
j = 1
while j < len(dst):
count = dst.count(dst[j])
if count >= 2:
result += count*(count-1)
j += count
return result
果然超出时间限制,看了看其他人的思路,其实跟自己的很像,但是不是存储的每两个点之间的距离,而是到一个点的距离为某个distance的点的数量,这样一来就不用每次通过count来获得这个值,时间复杂度就小了
代码:
class Solution:
def numberOfBoomerangs(self, points):
"""
:type points: List[List[int]]
:rtype: int
"""
def dis(point1, point2):
return (point1[0] - point2[0])**2 + (point1[1] - point2[1])**2
result = 0
for i in points:
record = {}
for j in points:
if j != i:
distance = dis(i, j)
record[distance] = record.get(distance, 0) + 1
for val in record.values():
if val >= 2:
result += (val - 1)* val
return result