「这是我参与11月更文挑战的第15天,活动详情查看:2021最后一次更文挑战」
题目
你打算用一个水罐给花园里的 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
输出:3
解释:从河边开始,此时水罐是装满的:
- 走到植物 0,1,2 (3 步) ,浇水。回到河边取水 (3 步)。
- 走到植物 3 (4 步) ,浇水。回到河边取水 (4 步)。
- 走到植物 4 (5 步) ,浇水。回到河边取水 (5 步)。
- 走到植物 5 (6 步) ,浇水。
需要的步数是 = 3 + 3 + 4 + 4 + 5 + 5 + 6 = 30 。
解题思路
题目即思路
提桶浇水,分析题目,假设当前在i位置;分析水桶的水;
因为能走到第i位置,当前水桶水量一定是大于浇花需要的水量的
分析浇完花后当前水桶剩余水量是否可以大于等于下一个花盆需要的水量;
如果浇完花后当前水桶剩余水量是否可以大于等于下一个花盆需要的水量;想前走一步;否则回去提水
计算回去提水需要花费的步数;当前位置i到池塘边需要走i步,返回到当前位置,需要i步;所以到下一个花盆需要步;
根据上述思路编辑代码如下:
如果当前
var wateringPlants = function (plants, capacity) {
const len = plants.length
let c = capacity
let result = 0
for (let i = 0; i < len; i++) {
let current = plants[i]
if (c >= current) {
c -= current
result += 1
}
if (i < len && c < plants[i + 1]) {
// 回去提水
result += 2 * (i + 1)
c = capacity
}
}
return result
}
解题思路分析
时间复杂度O(n);
空间复杂度:常量c