链接
题目
冬季已经来临。 你的任务是设计一个有固定加热半径的供暖器向所有房屋供暖。
在加热器的加热半径范围内的每个房屋都可以获得供暖。
现在,给出位于一条水平线上的房屋 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
提示:
1 <= houses.length, heaters.length <= 3 * 10^4
1 <= houses[i],heaters[i] <= 109
代码
class Solution {
public:
int findRadius(vector<int>& houses, vector<int>& heaters) {
sort(houses.begin(),houses.end());
sort(heaters.begin(),heaters.end());
int index = 0;
int i = 0;
int r = 0;
while(i < houses.size()){
//计算只有一个加湿器的情况,使用while循环直接全部解决;
//对于多个加湿器,有两种情况比较特殊:
//1.房子在最左边加湿器的左边;
//2.房子在最右边加湿器的右边;
//下面的while循环还可以解决最右边加湿器的右边;
//下面的if语句的作用:当每个房子的最近加湿器被算出来后就直接返回了;
while(index == heaters.size() - 1 && i < houses.size()){
int tmp = (heaters[index] - houses[i] > 0) ? (heaters[index] - houses[i]) : (houses[i] - heaters[index]);
r = max(r,tmp);
//cout << "i = "<<i << " : last tmp = " << tmp << ", min_r = "<<r << endl;
++i;
}
if(i >= houses.size()) break;
//下面的情况一定发生在多个加湿器,如果单个加湿器,在上面的while循环中已经if条件结束了。
//index迭代条件,当index处的加湿器要越过index+1处的加湿器加湿式,肯定不是最小加湿半径,要将加湿器的index提高到index+1是大于houses[i]的情况;
//特殊地,index已经是最后一个加湿器了,此时就不再迭代了,而是continue到index==heaters.size()-1的条件下去处理最右边加湿器的右边的房子的情况。
while(index < heaters.size() - 1 && heaters[index+1] <= houses[i]) {
++index;
}
if(index >= heaters.size() - 1){
continue;
}
//处理最左边加湿器左边的情况:if(houses[i] < heaters[index]);
//其他的就是在处理房子的左边和右边都有加湿器的情况:else。
if(houses[i] < heaters[index]){
r = max(r, heaters[index] - houses[i]);
}else{
int tmp1 = houses[i] - heaters[index];
int tmp2 = heaters[index+1] - houses[i];
r = max(r,min(tmp1,tmp2));
}
++i;
}
return r;
}
};
注意点
index的迭代
while(index < heaters.size() - 1 && heaters[index+1] <= houses[i]) {
++index;
}
作用:
这个迭代使用while循环,首先对于一个房子的位置此时大于index处的加湿器的位置,他会一直找到一个index满足下面的条件:
heaters[index] <= houses[i] < heaters[index+1]
warning:这里的代码不要写成:
while(heaters[index+1] <= houses[i] && index < heaters.size() - 1 ) {
++index;
}
此时++index
如果到最右边的加湿器,那么index+1
就会越界,因为&&
操作符有短路的性质,会先判断heaters[index+1] <= houses[i]
,这样就直接越界了,你放在后面,就会把这个判断短路!