ᕕ( ᐛ )ᕗ Jimyag's Blog

LeetCode-475-供暖器

题目

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

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

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

说明：所有供暖器都遵循你的半径标准，加热的半径也一样。

示例

1
2
3

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

1
2
3

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

1
2

输入：houses = [1,5], heaters = [2]
输出：3

解答

  1. 对于每个房屋，要么用前面的暖气，要么用后面的，二者取近的，得到距离；
  2. 对于所有的房屋，选择最大的上述距离。

代码

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17

int findRadius(vector<int> &houses, vector<int> &heaters) {
    sort(heaters.begin(), heaters.end());
    sort(houses.begin(), houses.end());
    int ans = 0;
    for (int house: houses) {
        // 找到比该房间右边的第一个暖气的索引
        int currentRightHeater = upper_bound(heaters.begin(), heaters.end(), house) - heaters.begin();
        int currentLeftHeater = currentRightHeater - 1;
        int rightDistance = currentRightHeater >= heaters.size() ? INT_MAX : heaters[currentRightHeater] - house;
        int leftDistance = currentLeftHeater < 0 ? INT_MAX : house - heaters[currentLeftHeater];
        // 他要用最近的一个这样才能保证最小，
        int curDistance = min(leftDistance, rightDistance);
        // 在所有的房屋都要满足要求，则要取最大的一个
        ans = max(ans, curDistance);
    }
    return ans;
}

#中等

目录