Leetcode 475. 供暖器
题目
冬季已经来临。 你的任务是设计一个有固定加热半径的供暖器向所有房屋供暖。
现在,给出位于一条水平线上的房屋和供暖器的位置,找到可以覆盖所有房屋的最小加热半径。
所以,你的输入将会是房屋和供暖器的位置。你将输出供暖器的最小加热半径。
说明:
- 给出的房屋和供暖器的数目是非负数且不会超过 25000。
- 给出的房屋和供暖器的位置均是非负数且不会超过10^9。
- 只要房屋位于供暖器的半径内(包括在边缘上),它就可以得到供暖。
- 所有供暖器都遵循你的半径标准,加热的半径也一样。
测试样例
示例 1:
输入: [1,2,3],[2]
输出: 1
解释: 仅在位置2上有一个供暖器。如果我们将加热半径设为1,那么所有房屋就都能得到供暖。
示例 2:
输入: [1,2,3,4],[1,4]
输出: 1
解释: 在位置1, 4上有两个供暖器。我们需要将加热半径设为1,这样所有房屋就都能得到供暖。
题解
对于每个房子,我们找到离它最近的一个热水器,我们就用这个作为我们的加热半径。要求每个房子都要得到供热,因此取刚才得到的结果的最大值。
对于查找最近的热水器,我们可以先对热水器进行排序,使用二分查找进行查找大于等于这个房子的一个热水器,那么这个热水器和它相邻(距离小于)的热水器有一个充当最近的热水器。详细过程见代码
代码
int findRadius(vector<int>& houses, vector<int>& heaters) {
sort(heaters.begin(),heaters.end());
int n1 = houses.size(), n2 = heaters.size();
int l = 0,r = n2-1,mid;
int ans=0;
for(int i=0; i<n1; i++){
l=0; r=n2-1;
while(l < r){ //二分查找,当前位置大于等于这个房子的第一个热水器
mid = (l+r)/2;
if(houses[i]<heaters[mid]){
r = mid;
}else{
l = mid + 1;
}
}
if(l != 0){ //考虑离房子最近的热水器
ans = max(ans,min(abs(heaters[l]-houses[i]),abs(houses[i]-heaters[l-1])));
}else ans = max(ans,abs(heaters[l]-houses[i]));
}
return ans;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/heaters
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。