思路1、贪心+双指针
首先排个序,题目没说保证升序,按坐标轴顺序排由小到大,用双指针分别指房子和加热器;
贪心的想法,加热半径最小,即离房子最近的两个加热器之间,距离的最小值;
如果满足这个加热半径,就往后看下一个,直到不满足加热半径且是这两个加热器离房子最近时更新答案;
初始值设为,如果第一个房子比第一个加热器小,设为两者距离(第一个加热器最近,只能由它供暖),否则为0(第一个房子可能由第二个供暖)。
具体实现看注释。
代码
class Solution {
public:
int findRadius(vector<int>& houses, vector<int>& heaters) {
sort(houses.begin(), houses.end());
sort(heaters.begin(), heaters.end());
int n = houses.size(), m = heaters.size();
if(m == 1) return max(abs(houses[0] - heaters[0]), abs(houses[n-1] - heaters[0]));
int ans = 0;
if(houses[0] <= heaters[0]) ans = heaters[0] - houses[0];
int i = 0, j = 0;
while(i < n && j < m) {
if(abs(houses[i] - heaters[j]) <= ans) //自己能管辖
i++;
else if(j + 1 < m) { //有下一个取暖器
if(abs(houses[i] - heaters[j+1]) <= ans) //下一个能管,归下一个管辖
i++, j++;
else {
if(houses[i] > heaters[j+1]) //房子更靠后,可能后面的取暖器更近,试探下一个
j++;
else { //只有当前两个离房子最近,更新ans为距离的最小值
ans = min(abs(houses[i] - heaters[j]), abs(houses[i] - heaters[j+1]));
if(abs(houses[i] - heaters[j]) > ans) //更新后不是由当前取暖器管,归下一个
i++, j++;
else i++;
}
}
}
else //没有下一取暖器且超范围,直接取最远的的房子和当前取暖器距离
return abs(houses[n-1] - heaters[j]);
}
return ans;
}
};
思路2、二分答案
本题理解为最小化最大值,取暖半径为所有取暖范围中的最大值,而目标是最小化这个值,这类问题都可以用二分答案的方法
首先确定上下界,下界为0,上界为房子的中的最大值,答案必然在上下界内,满足有序,可以二分答案来求解。
对于每个二分过程中的mid,如果满足半径要求,说明答案取大了或刚好,令r=mid,反之是答案取小了,令l = mid +1;
而每个check,就是判断取这样的半径能不能满足取暖要求。
代码
class Solution {
public:
bool check(vector<int>& houses, vector<int>& heaters, int x) {
int n = houses.size(), m = heaters.size();
for(int i = 0, j = 0; i < n; i++) {
while(j < m && abs(houses[i] - heaters[j]) > x) {
j++;
}
if(j == m) return false;
}
return true;
}
int findRadius(vector<int>& houses, vector<int>& heaters) {
sort(houses.begin(), houses.end());
sort(heaters.begin(), heaters.end());
int n = houses.size(), m = heaters.size();
if(m == 1) return max(abs(houses[0] - heaters[0]), abs(houses[n-1] - heaters[0]));
int l = 0, r = houses[n-1];
while(l < r) {
int mid = (l + r) >> 1;
if(check(houses, heaters, mid)) r = mid;
else l = mid + 1;
}
return r;
}
};