leetcode 475. 供暖器(Heaters)

Posted 2021-02-14 zhanzq1

目录

• 题目描述：
• 示例 1:
• 示例 2:
• 解法：

题目描述：

冬季已经来临。 你的任务是设计一个有固定加热半径的供暖器向所有房屋供暖。

现在，给出位于一条水平线上的房屋和供暖器的位置，找到可以覆盖所有房屋的最小加热半径。

所以，你的输入将会是房屋和供暖器的位置。你将输出供暖器的最小加热半径。

说明:

1. 给出的房屋和供暖器的数目是非负数且不会超过 25000。
2. 给出的房屋和供暖器的位置均是非负数且不会超过10⁹。
3. 只要房屋位于供暖器的半径内(包括在边缘上)，它就可以得到供暖。
4. 所有供暖器都遵循你的半径标准，加热的半径也一样。

示例 1:

输入: 
    [1,2,3],[2]
输出: 
    1
解释: 
    仅在位置2上有一个供暖器。如果我们将加热半径设为1，那么所有房屋就都能得到供暖。

示例 2:

输入: 
    [1,2,3,4],[1,4]
输出: 
    1
解释: 
    在位置1, 4上有两个供暖器。我们需要将加热半径设为1，这样所有房屋就都能得到供暖。

---

解法：

class Solution {
public:
    int findRadius(vector<int>& houses, vector<int>& heaters) {
        sort(houses.begin(), houses.end());
        sort(heaters.begin(), heaters.end());
        int sz1 = houses.size();
        int sz2 = heaters.size();
        vector<int> left(sz1, -1);
        vector<int> right(sz1, -1);
        int i = 0, j = 0;
        while(i < sz1){
            while(j < sz2 && heaters[j] <= houses[i]){
                j++;
            }
            if(j > 0){
                left[i] = heaters[j-1];
            }
            i++;
        }
        i = sz1-1, j = sz2-1;
        while(i >= 0){
            while(j >= 0 && heaters[j] >= houses[i]){
                j--;
            }
            if(j < sz2 - 1){
                right[i] = heaters[j+1];
            }
            i--;
        }
        int res = -1;
        for(int i = 0; i < sz1; i++){
            int tmp = 0;
            if(left[i] == -1){
                tmp = right[i] - houses[i];
            }else if(right[i] == -1){
                tmp = houses[i] - left[i];
            }else{
                tmp = min(houses[i] - left[i], right[i] - houses[i]);
            }
            res = max(res, tmp);
        }
        return res;
    }
};