LeetCode每日一题——1620. 网络信号最好的坐标

LeetCode每日一题系列

题目：1620. 网络信号最好的坐标
难度：普通

---

---

题目

给你一个数组 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,

1. 信号强度之和为 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]