一、题目
给你一个数组 towers
和一个整数 radius
。
数组 towers
中包含一些网络信号塔,其中 towers[i] = [xi, yi, qi]
表示第 i
个网络信号塔的坐标是 (xi, yi)
且信号强度参数为 qi
。所有坐标都是在 X-Y 坐标系内的 整数 坐标。两个坐标之间的距离用 欧几里得距离 计算。
整数 radius
表示一个塔 能到达 的 最远距离 。如果一个坐标跟塔的距离在 radius
以内,那么该塔的信号可以到达该坐标。在这个范围以外信号会很微弱,所以 radius
以外的距离该塔是 不能到达的 。
如果第 i
个塔能到达 (x, y)
,那么该塔在此处的信号为 ⌊qi / (1 + d)⌋
,其中 d
是塔跟此坐标的距离。一个坐标的 信号强度 是所有 能到达 该坐标的塔的信号强度之和。
请你返回数组 [cx, cy]
,表示 信号强度 最大的 整数 坐标点 (cx, cy)
。如果有多个坐标网络信号一样大,请你返回字典序最小的 非负 坐标。
二、思路
题中给出塔最大覆盖范围为50且坐标非负,可以一一遍历所有坐标点,求每一个坐标点的信号强度和,返回信号强度最大的坐标。
三、代码
class Solution:
def bestCoordinate(self, towers: List[List[int]], radius: int) -> List[int]:
mx = 0
ans = [0, 0]
for i in range(51):
for j in range(51):
t = 0
for x, y, q in towers:
d = ((x - i) ** 2 + (y - j) ** 2) ** 0.5
if d <= radius:
t += floor(q / (1 + d))
if t > mx:
mx = t
ans = [i, j]
return ans