Leetcode.1326 灌溉花园的最少水龙头数目

最新推荐文章于 2024-09-16 06:54:08 发布

感觉画质不如…原神

最新推荐文章于 2024-09-16 06:54:08 发布

阅读量293

点赞数 1

分类专栏： Leetcode 文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/m0_74396439/article/details/129140774

版权

Leetcode 专栏收录该内容

441 篇文章 2 订阅

订阅专栏

题目链接

Leetcode.1326 灌溉花园的最少水龙头数目 Rating ： 1885

题目描述

在 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
解释：即使打开所有水龙头，你也无法灌溉整个花园。

提示：

$1 <= n <= 10^4$
$r an g es . l e n g t h == n + 1$
$0 <= r an g es [i] <= 100$

分析：

在这里插入图片描述
我们用 rightEnd[i]记录以距离i点最远的右端点。

用 ans记录最终答案。
用 end记录到目前为止最远的右端点
用 prev记录上一个选择的最远的右端点

如果此时 end <= i $(0 <= i < n)$ ，说明还要区间覆盖不到，直接返回 -1。

如果此时 prev == i，说明又要选择一个新的区间了，ans++，prev = end。

时间复杂度： $O (n)$

C++代码：

class Solution {
public:
    int minTaps(int n, vector<int>& ranges) {
        vector<int> rightEnd(n+1);

        for(int i = 0;i <= n;i++){
           //我们只关心 [0,n]的情况，所以直接忽略负数的情况
            int left = max(0,i - ranges[i]);
            int right = i + ranges[i];
            rightEnd[left] = max(rightEnd[left],right);
        }

        int ans = 0,prev = 0,end = 0;
        for(int i = 0;i < n;i++){
            //每次都更新最大的右端点
            end = max(end,rightEnd[i]);
            //有区间不能被覆盖到
            if(end <= i) return -1;
            //之前的区间已经用完，现在要再添加一个新区间
            if(prev == i){
                ans++;
                prev = end;
            }
        }
        return ans;
    }
};

Java代码：

class Solution {
    public int minTaps(int n, int[] ranges) {
        int[] rightEnd = new int[n+1];

        for(int i = 0;i <= n;i++){
            int left = Math.max(0,i - ranges[i]);
            int right = i + ranges[i];
            rightEnd[left] = Math.max(rightEnd[left],right);
        }

        int ans = 0;
        int prev = 0;
        int end = 0;

        for(int i = 0;i < n;i++){
            end = Math.max(end,rightEnd[i]);

            if(end <= i) return -1;

            if(prev == i){
                ans++;
                prev = end;
            }
        }

        return ans;
    }
}