LeetCode-475 供暖器

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/heaters

题目描述

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

在加热器的加热半径范围内的每个房屋都可以获得供暖。

现在，给出位于一条水平线上的房屋 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 * 104
1 <= houses[i], heaters[i] <= 109

解题思路

解题的思路很显然，对于每一个房屋，都找到一个最近的加热器，计算屋子对于加热器的距离。

对于每一个房屋，首先对加热器位置排序，然后利用upper_bound接口找到第一个位置大于当前房屋的加热器位置。此时，有三种情况：

1、没有加热器位置比当前房屋大，那么就找位置最后的加热器求距离

2、找到加热器位置，但是加热器在最开始，那么求这个加热器与房屋的距离

3、找到加热器位置，但加热器不是在最开始的位置，比较这个加热器和上一个位置的加热器与房屋的位置，取最小。

代码展示

class Solution {
public:
    int findRadius(vector<int>& houses, vector<int>& heaters) {
        int iRet = 0;
        sort(heaters.begin(), heaters.end());
        for(auto house: houses)
        {
            int  iIndex = upper_bound(heaters.begin(), heaters.end(), house) - heaters.begin();
            if(iIndex == heaters.size())
            {
                iRet = max(iRet, house - heaters[iIndex - 1]);
            }
            else
            {
                int iMinR;
                if(iIndex > 0)
                {
                    iMinR = min(heaters[iIndex] - house, house - heaters[iIndex - 1]);
                }
                else
                {
                    iMinR = heaters[iIndex] - house;
                }
                iRet = max(iRet, iMinR);
            }
        }
        return iRet;
    }
};

运行成果