Leetcode 1326:灌溉花园的最少水龙头数目(超详细的解法!!!)

在 x 轴上有一个一维的花园。花园长度为 n,从点 0 开始,到点 n 结束。

花园里总共有 n + 1 个水龙头,分别位于 [0, 1, ..., n]

给你一个整数 n 和一个长度为 n + 1 的整数数组 ranges ,其中 ranges[i] (下标从 0 开始)表示:如果打开点 i 处的水龙头,可以灌溉的区域为 [i - ranges[i], i + ranges[i]]

请你返回可以灌溉整个花园的 最少水龙头数目 。如果花园始终存在无法灌溉到的地方,请你返回 -1

示例 1:

输入:n = 5, ranges = [3,4,1,1,0,0]
输出:1
解释:
点 0 处的水龙头可以灌溉区间 [-3,3]
点 1 处的水龙头可以灌溉区间 [-3,5]
点 2 处的水龙头可以灌溉区间 [1,3]
点 3 处的水龙头可以灌溉区间 [2,4]
点 4 处的水龙头可以灌溉区间 [4,4]
点 5 处的水龙头可以灌溉区间 [5,5]
只需要打开点 1 处的水龙头即可灌溉整个花园 [0,5] 。

示例 2:

输入:n = 3, ranges = [0,0,0,0]
输出:-1
解释:即使打开所有水龙头,你也无法灌溉整个花园。

示例 3:

输入:n = 7, ranges = [1,2,1,0,2,1,0,1]
输出:3

示例 4:

输入:n = 8, ranges = [4,0,0,0,0,0,0,0,4]
输出:2

示例 5:

输入:n = 8, ranges = [4,0,0,0,4,0,0,0,4]
输出:1

提示:

  • 1 <= n <= 10^4
  • ranges.length == n + 1
  • 0 <= ranges[i] <= 100

解题思路

这个问题实际上和Leetcode 1024:视频拼接(超详细的解法!!!)一模一样。为了可以直接套用之前的代码,我们建立一个跳跃数组即可。

class Solution:
    def minTaps(self, n: int, ranges: List[int]) -> int:
        data = [0] * (n + 1)
        for i, v in enumerate(ranges):
            l, r = max(0, i - v), min(n, i + v)
            data[l] = max(data[l], r - l)
        
        pre, cur, step = -1, 0, 0
        for i in range(n + 1):
            if cur < i:
                return -1
            if cur >= n: 
                return step
            elif pre < i <= cur:
                step, pre = step + 1, cur
            cur = max(cur, i + data[i])     
        return step

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题,希望大家指出!!!

发布了731 篇原创文章 · 获赞 457 · 访问量 83万+

猜你喜欢

转载自blog.csdn.net/qq_17550379/article/details/104060060