题目
给你一个数组 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) 。如果有多个坐标网络信号一样大,请你返回字典序最小的 非负 坐标。
注意:
坐标 (x1, y1) 字典序比另一个坐标 (x2, y2) 小,需满足以下条件之一:
要么 x1 < x2 ,
要么 x1 == x2 且 y1 < y2 。
⌊val⌋ 表示小于等于 val 的最大整数(向下取整函数)。
示例
示例 1:
输入:towers = [[1,2,5],[2,1,7],[3,1,9]], radius = 2
输出:[2,1] 解释: 坐标 (2,
信号强度之和为 13
塔 (2, 1) 强度参数为 7 ,在该点强度为 ⌊7 / (1 + sqrt(0)⌋ = ⌊7⌋ = 7
塔 (1, 2) 强度参数为 5 ,在该点强度为 ⌊5 / (1 + sqrt(2)⌋ = ⌊2.07⌋ = 2
塔 (3, 1) 强度参数为 9 ,在该点强度为 ⌊9 / (1 + sqrt(1)⌋ = ⌊4.5⌋ = 4 没有别的坐标有更大的信号强度。
示例 2:
输入:towers = [[23,11,21]], radius = 9
输出:[23,11]
解释:由于仅存在一座信号塔,所以塔的位置信号强度最大。
示例 3:
输入:towers = [[1,2,13],[2,1,7],[0,1,9]], radius = 2
输出:[1,2]
解释:坐标 (1,2) 的信号强度最大。
提示:
1 <= towers.length <= 50
towers[i].length == 3
0 <= xi, yi, qi <= 50
1 <= radius <= 50
思路
首先看这个数据范围第一想法是暴力搜索所有可能的点的坐标,具体做法是:
1.遍历一遍给定towers,找到所有信号塔最大的x坐标x_max和最大的y坐标y_max
2.从零开始遍历x_max和y_max,在每个x,y中累计towers中的信号塔能否到达,以及对该坐标信号的贡献
3.维持信号贡献的最大值,因为坐标是按照x,y递增的顺序遍历的,所以最后遇到的信号最大值就是字典序最大的坐标,返回即可
题解
class Solution: def bestCoordinate(self, towers: List[List[int]], radius: int) -> List[int]: # 求出x,y最大坐标 x_max = max(t[0] for t in towers) y_max = max(t[1] for t in towers) cx = cy = max_quality = 0 # 从0遍历x_max,y_max for x in range(x_max + 1): for y in range(y_max + 1): quality = 0 # 累计每个信号塔的贡献 for tx, ty, q in towers: d = (x - tx) ** 2 + (y - ty) ** 2 if d <= radius ** 2: quality += int(q / (1 + d ** 0.5)) # 维持信号贡献最大值,以及对应坐标 if quality > max_quality: cx, cy, max_quality = x, y, quality # 最后返回即可 return [cx, cy]