前端算法第一一五弹-供暖器

76 阅读3分钟

「这是我参与2022首次更文挑战的第22天,活动详情查看:2022首次更文挑战

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

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

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

排序 + 二分查找

为了使供暖器可以覆盖所有房屋且供暖器的加热半径最小,对于每个房屋,应该选择离该房屋最近的供暖器覆盖该房屋,最近的供暖器和房屋的距离即为该房屋需要的供暖器的最小加热半径。所有房屋需要的供暖器的最小加热半径中的最大值即为可以覆盖所有房屋的最小加热半径。

为了得到距离每个房屋最近的供暖器,可以将供暖器数组 heatersheaters 排序,然后通过二分查找得到距离最近的供暖器。

具体而言,对于每个房屋 househouse,需要在有序数组 heatersheaters 中找到最大的下标 i,使得 heaters[i]househeaters[i]≤house,特别地,当 heaters[0]>househeaters[0]>house 时,i=1i=−1。在得到下标 i 之后,令 j=i+1j=i+1,则 j 是满足 heaters[j]>househeaters[j]>house 的最小下标。特别地,当 heaters[n1]househeaters[n−1]≤house 时,j=nj=n,其中 n 是数组 heatersheaters 的长度。

得到下标 i 和 j 之后,离房屋 househouse 最近的供暖器为 heaters[i]heaters[i]heaters[j]heaters[j],分别计算这两个供暖器和房屋 househouse 的距离,其中的最小值即为该房屋需要的供暖器的最小加热半径。对于 i=1i=−1j=nj=n 的下标越界情况,只要将对应的距离设为 ++∞ 即可。

var findRadius = function(houses, heaters) {
    let ans = 0;
    heaters.sort((a, b) => a - b);
    for (const house of houses) {
        const i = binarySearch(heaters, house);
        const j = i + 1;
        const leftDistance = i < 0 ? Number.MAX_VALUE : house - heaters[i];
        const rightDistance = j >= heaters.length ? Number.MAX_VALUE : heaters[j] - house;
        const curDistance = Math.min(leftDistance, rightDistance);
        ans = Math.max(ans, curDistance);
    }
    return ans;
};

const binarySearch = (nums, target) => {
    let left = 0, right = nums.length - 1;
    if (nums[left] > target) {
        return -1;
    }
    while (left < right) {
        const mid = Math.floor((right - left + 1) / 2) + left;
        if (nums[mid] > target) {
            right = mid - 1;
        } else {
            left = mid;
        }
    }
    return left;
}

}

复杂度分析

  • 时间复杂度:O((n+m)logn)O((n+m)logn),其中 m 是数组 houseshouses 的长度,n 是数组 heatersheaters 的长度。对数组 heatersheaters 排序需要 O(nlogn)O(nlog⁡n) 的时间。

    使用二分查找对每个房屋寻找距离最近的供暖器,每次二分查找需要 O(logn)O(log⁡n) 的时间,有 m 个房屋,因此需要 O(mlogn)O(mlog⁡n) 的时间。

    总时间复杂度是 O((n+m)logn)O((n+m)log⁡n)

  • 总时间复杂度是 O(logn)O(log⁡n),其中 n 是数组 heatersheaters 的长度。空间复杂度主要取决于排序所需要的空间。