X

LEETCODE 2079. Watering Plants 解题思路分析

题目大意:

给植物浇水

你打算用一个水罐给花园里的 n 株植物浇水。植物排成一行,从左到右进行标记,编号从 0 到 n – 1 。其中,第 i 株植物的位置是 x = i 。x = -1 处有一条河,你可以在那里重新灌满你的水罐。

每一株植物都需要浇特定量的水。你将会按下面描述的方式完成浇水:

  • 按从左到右的顺序给植物浇水。
  • 在给当前植物浇完水之后,如果你没有足够的水 完全 浇灌下一株植物,那么你就需要返回河边重新装满水罐。
  • 你 不能 提前重新灌满水罐。

最初,你在河边(也就是,x = -1),在 x 轴上每移动 一个单位 都需要 一步

给你一个下标从 0 开始的整数数组 plants ,数组由 n 个整数组成。其中,plants[i] 为第 i 株植物需要的水量。另有一个整数 capacity 表示水罐的容量,返回浇灌所有植物需要的 步数 。

示例 1:

输入:plants = [2,2,3,3], capacity = 5
输出:14
解释:从河边开始,此时水罐是装满的:
- 走到植物 0 (1 步) ,浇水。水罐中还有 3 单位的水。
- 走到植物 1 (1 步) ,浇水。水罐中还有 1 单位的水。
- 由于不能完全浇灌植物 2 ,回到河边取水 (2 步)。
- 走到植物 2 (3 步) ,浇水。水罐中还有 2 单位的水。
- 由于不能完全浇灌植物 3 ,回到河边取水 (3 步)。
- 走到植物 3 (4 步) ,浇水。
需要的步数是 = 1 + 1 + 2 + 3 + 3 + 4 = 14 。

示例 2:

输入:plants = [1,1,1,4,2,3], capacity = 4
输出:30
解释:从河边开始,此时水罐是装满的:
- 走到植物 0,1,2 (3 步) ,浇水。回到河边取水 (3 步)。
- 走到植物 3 (4 步) ,浇水。回到河边取水 (4 步)。
- 走到植物 4 (5 步) ,浇水。回到河边取水 (5 步)。
- 走到植物 5 (6 步) ,浇水。
需要的步数是 = 3 + 3 + 4 + 4 + 5 + 5 + 6 = 30 。

示例 3:

输入:plants = [7,7,7,7,7,7,7], capacity = 8
输出:49
解释:每次浇水都需要重新灌满水罐。
需要的步数是 = 1 + 1 + 2 + 2 + 3 + 3 + 4 + 4 + 5 + 5 + 6 + 6 + 7 = 49 。

提示:

n == plants.length
1 <= n <= 1000
1 <= plants[i] <= 106
max(plants[i]) <= capacity <= 109


解题思路分析:

这道题没有什么难度,按照步骤算就可以。我们使用一个前缀和累积每次打满水后到当前位置需要的水量,如果水量足够,无需返回补水,步数加一即可。如果水量不够时,需要返回河边打水,所需步数也就是当前位置到河边一来一回的距离和,这里需要注意一点,比如到第5颗植物时水不够了,我们不用走到第五颗植物后再返回河边,而是在浇灌完第4颗植物后返回即可,因此步数应该是第五颗植物距离河边距离(5)乘以2减一。补水后返回到需要灌溉的植物,此时前缀和变为当前植物需要的水量。

实现代码:

class Solution {
    public int wateringPlants(int[] plants, int capacity) {
        int res=0; // 返回结果
        int preSum=0; // 补水后到当前植物为止消耗的水量
        for(int i=0;i<plants.length;i++){ // 循环每个植物
            preSum+=plants[i]; // 累加当前需要消耗的水量
            if(preSum<=capacity){ // 水量充足时,
                res++; // 直接灌溉,步数加一
            }else{ // 水量不足时
                res+=(i+1)*2-1;  // 返回河边取水
                preSum = plants[i]; // 消耗水量为当前植物需要的水量
            }
        }
        return res;
    }
}

本题解法执行时间为0ms。

Runtime: 0 ms, faster than 100.00% of Java online submissions for Watering Plants.Memory Usage: 38.8 MB, less than 20.00% of Java online submissions for Watering Plants.

本网站文章均为原创内容,并可随意转载,但请标明本文链接
如有任何疑问可在文章底部留言。为了防止恶意评论,本博客现已开启留言审核功能。但是博主会在后台第一时间看到您的留言,并会在第一时间对您的留言进行回复!欢迎交流!
本文链接: https://leetcode.jp/leetcode-2079-watering-plants-解题思路分析/
Categories: leetcode
kwantong: