携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第3天，点击查看活动详情
题目
leetcode 2079. 给植物浇水
你打算用一个水罐给花园里的 n 株植物浇水。植物排成一行，从左到右进行标记，编号从 0 到 n - 1 。其中，第 i 株植物的位置是 x = i 。x = -1 处有一条河，你可以在那里重新灌满你的水罐。 每一株植物都需要浇特定量的水。你将会按下面描述的方式完成浇水：
按从左到右的顺序给植物浇水。 在给当前植物浇完水之后，如果你没有足够的水 完全 浇灌下一株植物，那么你就需要返回河边重新装满水罐。 你 不能 提前重新灌满水罐。 最初，你在河边（也就是，x = -1），在 x 轴上每移动 一个单位 都需要 一步 。 给你一个下标从 0 开始的整数数组 plants ，数组由 n 个整数组成。其中，plants[i] 为第 i 株植物需要的水量。另有一个整数 capacity 表示水罐的容量，返回浇灌所有植物需要的步数 。
示例：
输入：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 。
题解
题目难度为中等难度，但是实际上还是很容易写的，所以大家在刷leetcode的时候也不用怕被题目吓到，认真分析写起来也不会很难。根据题目意思我们只需要按照数组的顺序依次浇灌，关注水罐剩余水量和下一次需要灌溉的水量。
var wateringPlants = function(plants, capacity) {
let cur = capacity;
let res = 0;
for (let i = 0; i < plants.length; i++) {
cur -= plants[i];
res++;
if (cur < plants[i + 1]) {
cur = capacity;
res += (i + 1) * 2;
}
}
return res
};
代码详解
定义变量cur表示水罐当前剩余水量，res表示当前水量。
循环数组plants，每前进一步水罐就减去浇掉的水量。如果下一次的水量大于罐中的水量则需要返回取水。其中步数为往返两次的路程。
