【LeetCode】475. 供暖器

题目描述

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

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

现在，给出位于一条水平线上的房屋 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

链接：https://leetcode-cn.com/problems/heaters

打卡

思路

• 遍历每一个house，每一轮找出最小的dis使得满足这个house，然后对这些最小的dis取最大值，即可以满足所有的house都能覆盖。
• 先让heaters排序，每一轮使用二分法算出最小的dis。
  • 尝试找到正好等于house的middle坐标，当可以找到，说明使用这个middle位置的供暖期即可满足覆盖，此时dis= 0；
  • 当house在heaters所有元素的左边，此时right不停地向左边移动，当等于0的时候退出循环，此时用第一个供暖器即可满足这个house的覆盖，所以dis = heaters[0] - house;
  • 当house 在heaters所有元素的右边，类似地dis = house - heaters[n-1];
  • 当house 在heaters 2个元素之间（这个最难，经常调试溢出），取和这两个元素的最小距离：dis = min(house - heaters[right-1] , heaters[left] - house)。//dis = min(house - heaters[right-1] , heaters[right] - house);也是对的，因为此时 left == right，并且位置在house的右边。

代码

class Solution {
public:
    int findRadius(vector<int>& houses, vector<int>& heaters) {
        sort(heaters.begin(), heaters.end());
        int n = heaters.size();
        int res = INT_MIN;
        int dis = INT_MAX;
        for(auto& house: houses)
        {
            int left = 0;
            int right = n;
            bool found = false;
            while(left < right)
            {
                int middle = left + (right - left)/2;
                if(heaters[middle] > house)
                {
                    right = middle; 
                }
                else if(heaters[middle] < house)
                {
                    left = middle + 1;
                }
                else
                {
                    found = true;
                    break;
                }
            }

            if(found)
            {
                dis = 0;
            }
            else if(right == 0)  // house 在heaters所有元素的左边
            {
                dis = heaters[0] - house;
            }
            else if(left == n) //  house 在heaters所有元素的右边
            {
                dis = house - heaters[n-1];
            }
            else   //  house 在heaters 元素中间 heaters[right -1], house(right), heaters[left] 
            {
                dis = min(house - heaters[right-1] , heaters[left] - house);
                //dis = min(house - heaters[right-1] , heaters[right] - house);
            }
            res = max(res, dis);
        }

        return res;
    }
};