题目描述
冬季已经来临。 你的任务是设计一个有固定加热半径的供暖器向所有房屋供暖。
在加热器的加热半径范围内的每个房屋都可以获得供暖。
现在,给出位于一条水平线上的房屋 houses 和供暖器 heaters 的位置,请你找出并返回可以覆盖所有房屋的最小加热半径。
说明:所有供暖器都遵循你的半径标准,加热的半径也一样。
示例 1:
输入: houses = [1,2,3], heaters = [2]
输出: 1
解释: 仅在位置2上有一个供暖器。如果我们将加热半径设为1,那么所有房屋就都能得到供暖。
示例 2:
输入: houses = [1,2,3,4], heaters = [1,4]
输出: 1
解释: 在位置1, 4上有两个供暖器。我们需要将加热半径设为1,这样所有房屋就都能得到供暖。
示例 3:
输入:houses = [1,5], heaters = [2]
输出:3
链接:https://leetcode-cn.com/problems/heaters
打卡
思路
- 遍历每一个house,每一轮找出最小的dis使得满足这个house,然后对这些最小的dis取最大值,即可以满足所有的house都能覆盖。
- 先让heaters排序,每一轮使用二分法算出最小的dis。
- 尝试找到正好等于house的middle坐标,当可以找到,说明使用这个middle位置的供暖期即可满足覆盖,此时dis= 0;
- 当house在heaters所有元素的左边,此时right不停地向左边移动,当等于0的时候退出循环,此时用第一个供暖器即可满足这个house的覆盖,所以dis = heaters[0] - house;
- 当house 在heaters所有元素的右边,类似地dis = house - heaters[n-1];
- 当house 在heaters 2个元素之间(这个最难,经常调试溢出),取和这两个元素的最小距离:dis = min(house - heaters[right-1] , heaters[left] - house)。//dis = min(house - heaters[right-1] , heaters[right] - house);也是对的,因为此时 left == right,并且位置在house的右边。
代码
class Solution {
public:
int findRadius(vector<int>& houses, vector<int>& heaters) {
sort(heaters.begin(), heaters.end());
int n = heaters.size();
int res = INT_MIN;
int dis = INT_MAX;
for(auto& house: houses)
{
int left = 0;
int right = n;
bool found = false;
while(left < right)
{
int middle = left + (right - left)/2;
if(heaters[middle] > house)
{
right = middle;
}
else if(heaters[middle] < house)
{
left = middle + 1;
}
else
{
found = true;
break;
}
}
if(found)
{
dis = 0;
}
else if(right == 0) // house 在heaters所有元素的左边
{
dis = heaters[0] - house;
}
else if(left == n) // house 在heaters所有元素的右边
{
dis = house - heaters[n-1];
}
else // house 在heaters 元素中间 heaters[right -1], house(right), heaters[left]
{
dis = min(house - heaters[right-1] , heaters[left] - house);
//dis = min(house - heaters[right-1] , heaters[right] - house);
}
res = max(res, dis);
}
return res;
}
};