题目
难度:★★☆☆☆
类型:数组
冬季已经来临。 你的任务是设计一个有固定加热半径的供暖器向所有房屋供暖。
现在,给出位于一条水平线上的房屋和供暖器的位置,找到可以覆盖所有房屋的最小加热半径。
所以,你的输入将会是房屋和供暖器的位置。你将输出供暖器的最小加热半径。
说明:
给出的房屋和供暖器的数目是非负数且不会超过 25000。
给出的房屋和供暖器的位置均是非负数且不会超过10^9。
只要房屋位于供暖器的半径内(包括在边缘上),它就可以得到供暖。
所有供暖器都遵循你的半径标准,加热的半径也一样。
示例
示例 1:
输入: [1,2,3],[2]
输出: 1
解释: 仅在位置2上有一个供暖器。如果我们将加热半径设为1,那么所有房屋就都能得到供暖。
示例 2:
输入: [1,2,3,4],[1,4]
输出: 1
解释: 在位置1, 4上有两个供暖器。我们需要将加热半径设为1,这样所有房屋就都能得到供暖。
解答
为了找到保证所有房屋都得到供暖的暖器辐射范围,我们需要考虑房屋与暖器的最远距离即可。
对于每一个房屋,我们找到距离这个房屋最近的暖器,并算出距离dist;
对于所有房屋计算出的dist,取其中的最大值,即为暖器至少需要的辐射范围。
这里,我们使用二分搜索来确定一个数字在排序列表中的位置,调用Python中的bisect模块实现。例如,找到3.2在数组[1,2,3,4,5]中的位置,只需要bisect.bisect_left([1,2,3,4,5], 3.2)即可,函数返回3。
import bisect
class Solution:
def findRadius(self, houses, heaters):
res = 0
heaters.sort()
for house in sorted(houses):
idx = bisect.bisect_left(heaters, house)
if idx == 0: # 房屋house在所有暖器的左边
dist = heaters[0] - house # 离该房屋最近的暖器就是最左边的暖器
elif idx == len(heaters): # 房屋house在所有暖器的右边
dist = house - heaters[-1] # 离该房屋最近的暖器就是最右边的暖器
else: # 房屋house在暖器中间
dist = min(house-heaters[idx-1], heaters[idx]-house) # 房屋house左右最近暖器距离最小值
res = max(res, dist) # res用来统计最近距离的最大值
return res
如有疑问或建议,欢迎评论区留言~