看了题解之后在纸上画了画才写出来的。可以假设对于每个house左右都有加热器,那么这个house的半径就是min(左加热器,右加热器)。ans则是所有house半径的最大值。
初始化ans为0。将heaters从小到大排序,遍历houses,对于houses中的每个house,找到第1个大于house的位置:
1. 如果这个位置index等于0,说明heater的第1个数就大于house 这个house就只有这1个加热器(右加热器),这个半径就等于
,计算ans = max(ans, 这个半径)
2. 如果这个位置index等于heaters.size(),说明找到最后也没有在heaters中找到大于house的heater 这个house就只有最后一个加热器离它最近(左加热器),这个半径就等于
,取ans和这个半径的最大值
3. 然后就是一般情况,house的右边加热器为,house左边的加热器为
,这个house的最小半径就是
,这个最小半径和ans取最大值。
4. 最后返回ans即可
class Solution {
public:
int findRadius(vector<int>& houses, vector<int>& heaters) {
sort(heaters.begin(),heaters.end());
int ans = 0;
for (int house:houses){
// 在heaters中查找第1个大于house的位置
int uppos = upper_bound(heaters.begin(),heaters.end(),house) - heaters.begin();
if (uppos ==0 ){
// 只有一个加热器
ans = max(ans,heaters[uppos]-house);
}
else if (uppos == heaters.size()){
ans = max(ans,house - heaters[uppos-1]);
}
else{
// 有2个加热器
int lowpos = uppos-1;
int tmpans = min(house-heaters[lowpos],heaters[uppos]-house);
ans = max(ans,tmpans);
}
}
return ans;
}
};