【面试高频题】难度 1.5/5,常见的二分双指针面试题

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第15天,点击查看活动详情

题目描述

这是 LeetCode 上的 475. 供暖器 ,难度为 中等

Tag : 「排序」、「二分」、「双指针」

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

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

现在,给出位于一条水平线上的房屋 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 < = h o u s e s . l e n g t h , h e a t e r s . l e n g t h < = 3 1 0 4 1 <= houses.length, heaters.length <= 3 * 10^4
  • 1 < = h o u s e s [ i ] , h e a t e r s [ i ] < = 1 0 9 1 <= houses[i], heaters[i] <= 10^9

二分 + 双指针

需要求得最小加热半径 a n s ans ,使得所有的 h o u s e s [ i ] houses[i] 均被覆盖。

在以 a n s ans 为分割点的数轴上具有「二段性」:

  • 数值小于 a n s ans 的半径无法覆盖所有的房子;
  • 数值大于等于 a n s ans 的半径可以覆盖所有房子。

因此可直接「二分答案」,考虑应该在什么范围内进行「二分」。

可以从数据范围入手,使用 1 e 9 1e9 为二分上界,该做法能确保答案在二分范围内。

考虑如何实现 check 函数。

先对 h o u s e s houses h e a t e r s heaters 进行排序,使用 i i 指向当前处理到的 h o u s e s [ i ] houses[i] j j 指向 可能 覆盖到 h o u s e s [ i ] houses[i] 的最小下标 h e a t e r s [ j ] heaters[j] x x 代表当前需要 check 的半径。

当且仅当 h e a t e r s [ j ] + x < h o u s e s [ i ] heaters[j] + x < houses[i] 时, h o u s e s [ i ] houses[i] 必然不能被 h e a t e r s [ j ] heaters[j] 所覆盖,此时让 j j 自增。

找到合适的 j j 之后,再检查 h e a t e r s [ j ] x < = h o u s e s [ i ] < = h e a t e r s [ j ] + x heaters[j] - x <= houses[i] <= heaters[j] + x 是否满足,即可知道 h o u s e s [ i ] houses[i] 的覆盖情况。

代码:

class Solution {
    public int findRadius(int[] houses, int[] heaters) {
        Arrays.sort(houses);
        Arrays.sort(heaters);
        int l = 0, r = (int) 1e9;
        while (l < r) {
            int mid = l + r >> 1;
            if (check(houses, heaters, mid)) r = mid;
            else l = mid + 1;
        }
        return r;
    }
    boolean check(int[] houses, int[] heaters, int x) {
        int n = houses.length, m = heaters.length;
        for (int i = 0, j = 0; i < n; i++) {
            while (j < m && houses[i] > heaters[j] + x) j++;
            if (j < m && heaters[j] - x <= houses[i] && houses[i] <= heaters[j] + x) continue;
            return false;
        }
        return true;
    }
}
复制代码
  • 时间复杂度:令 n n m m 分别为 housesheaters 长度, L = 1 e 9 L = 1e9 为最大长度,对其进行排序复杂度为 O ( n log n + m log m ) O(n\log{n} + m\log{m}) ,在 [ 0 , L ] [0, L] 范围进行二分,单次 check 会使用「双指针」判断是否每个 h o u s e s [ i ] houses[i] 是否被覆盖,复杂度为 O ( max ( n , m ) log L ) O(\max(n, m) * \log{L}) 。整体复杂度为 O ( max ( n , m ) log L ) O(\max(n, m) * \log{L})
  • 空间复杂度:排序所需要消耗的空间。复杂度为 O ( log n + log m ) O(\log{n} + \log{m})

最后

这是我们「刷穿 LeetCode」系列文章的第 No.475 篇,系列开始于 2021/01/01,截止于起始日 LeetCode 上共有 1916 道题目,部分是有锁题,我们将先把所有不带锁的题目刷完。

在这个系列文章里面,除了讲解解题思路以外,还会尽可能给出最为简洁的代码。如果涉及通解还会相应的代码模板。

为了方便各位同学能够电脑上进行调试和提交代码,我建立了相关的仓库:github.com/SharingSour…

在仓库地址里,你可以看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其他优选题解。

猜你喜欢

转载自juejin.im/post/7106338215967064071