1. 题目
冬季已经来临。 你的任务是设计一个有固定加热半径的供暖器向所有房屋供暖。
现在,给出位于一条水平线上的房屋和供暖器的位置,找到可以覆盖所有房屋的最小加热半径。
所以,你的输入将会是房屋和供暖器的位置。你将输出供暖器的最小加热半径。
说明:
给出的房屋和供暖器的数目是非负数且不会超过 25000。
给出的房屋和供暖器的位置均是非负数且不会超过10^9。
只要房屋位于供暖器的半径内(包括在边缘上),它就可以得到供暖。
所有供暖器都遵循你的半径标准,加热的半径也一样。
示例 1:
输入: [1,2,3],[2]
输出: 1
解释: 仅在位置2上有一个供暖器。如果我们将加热半径设为1,那么所有房屋就都能得到供暖。
示例 2:
输入: [1,2,3,4],[1,4]
输出: 1
解释: 在位置1, 4上有两个供暖器。我们需要将加热半径设为1,这样所有房屋就都能得到供暖。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/heaters
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
2.1 双指针
- 将两个数组都排序(为方便处理边界,在排序前,向供热站里加入正负极限值)
- 设置两个指针指向房屋的起点,和供热站的起点
- 如果房屋在两个供热站之间,取较小的距离,最后答案是所有的较小距离的最大值
class Solution {
public:
int findRadius(vector<int>& houses, vector<int>& heaters) {
sort(houses.begin(),houses.end());
//为处理边界方便,增加两个虚拟无穷远加热站
heaters.push_back(INT_MAX);
heaters.push_back(-1e9-1);
sort(heaters.begin(),heaters.end());
int i = 0, j = 0, r = 0;
while(i < houses.size())
{
if(houses[i] >= heaters[j] && houses[i] <= heaters[j+1])
{
r = max(r,min(houses[i]-heaters[j], heaters[j+1]-houses[i]));
++i;
}
else
++j;
}
return r;
}
};
2. 二分查找
class Solution {
public:
int findRadius(vector<int>& houses, vector<int>& heaters) {
heaters.push_back(INT_MAX);
heaters.push_back(-1e9-1);
sort(heaters.begin(),heaters.end());
int i = 0, prev, next, r = 0;
for(i = 0; i < houses.size(); ++i)
{
prev = 0; next = heaters.size()-1;
binarysearch(houses[i],heaters,prev,next);//得到的prev,next为houses[i]后面和前面的供热站位置
r = max(r,min(houses[i]-heaters[next], heaters[prev]-houses[i]));
}
return r;
}
void binarysearch(int p, vector<int> &heaters, int &i, int &j)
{
int mid;
while(i <= j)
{
mid = i+((j-i)>>1);
if(p < heaters[mid])
j = mid-1;
else if(p > heaters[mid])
i = mid+1;
else
{
i = j = mid;
return;
}
}
}
};