題目描述
冬季已經來臨。 你的任務是設計一個有固定加熱半徑的供暖器向所有房屋供暖。
現在,給出位於一條水平線上的房屋和供暖器的位置,找到可以覆蓋所有房屋的最小加熱半徑。
所以,你的輸入將會是房屋和供暖器的位置。你將輸出供暖器的最小加熱半徑。
說明:
給出的房屋和供暖器的數目是非負數且不會超過 25000。
給出的房屋和供暖器的位置均是非負數且不會超過10^9。
只要房屋位於供暖器的半徑內(包括在邊緣上),它就可以得到供暖。
所有供暖器都遵循你的半徑標準,加熱的半徑也一樣。
示例 1:
輸入: [1,2,3],[2]
輸出: 1
解釋: 僅在位置2上有一個供暖器。如果我們將加熱半徑設爲1,那麼所有房屋就都能得到供暖。
示例 2:
輸入: [1,2,3,4],[1,4]
輸出: 1
解釋: 在位置1, 4上有兩個供暖器。我們需要將加熱半徑設爲1,這樣所有房屋就都能得到供暖。
算法思路
這兩個示例讓我以爲,所有房子都是挨着的,那麼house是連續的列表。
然後,所有房子都要供暖,很容易想到取相鄰供暖器的位置的最大差值,取二分之一即可。
但是。
class Solution:
def findRadius(self, houses: List[int], heaters: List[int]) -> int:
i,MAx=1,0
if len(heaters)==1:
return max(houses[-1]-heaters[0],heaters[0]-1)
while i<len(heaters):
tmp=heaters[i]-heaters[i-1]-1
MAx=max(MAx,tmp)
i+=1
return MAx//2 if MAx%2==0 else MAx//2+1
測試1:[1,5] [2]
測試2:[1,2,3,5,15] [2,30] 供暖器位置居然還有超出房屋的……誰裝的供暖器?打死好了,本來按題意求加熱半徑已經有重疊了,現在還把加熱器安在外面,奢靡浪費,打死。
哇,測例太噁心了,連參數都不排序的。
算法
給兩個參數排序,遍歷一遍房屋,計算並取每個房屋距離最近的取暖器距離。
對所有房屋的最近距離,取其中的最大值。
class Solution:
def findRadius(self, houses, heaters) -> int:
houses.sort()
heaters.sort()
j, MAx = 0, 0
for i in range(len(houses)):
while houses[i] > heaters[j]:
if j == len(heaters) - 1: break
j += 1
if j == 0:
MIN = abs(heaters[j] - houses[i])
else:
MIN = min(abs(heaters[j] - houses[i]), abs(houses[i] - heaters[j-1]))
MAx = max(MAx, MIN)
return MAx
執行用時 :332 ms, 在所有 Python3 提交中擊敗了84.19%的用戶
內存消耗 :16.5 MB, 在所有 Python3 提交中擊敗了9.87%的用戶