JZX轻语：简
LeetCode 2079 - 给植物浇水
发表于2024年05月08日
模拟浇水的流程即可，使用一个变量维护当前的水量，如果当前的植物浇水量大于当前水量，则加上折返的步数，否则加1即可。
直观的做法：
class Solution:
def wateringPlants(self, plants: List[int], capacity: int) -> int:
cur_cap = capacity
ans = 0
for i, plant in enumerate(plants):
ans += 1
if cur_cap < plant:
ans += 2 * i
cur_cap = capacity
cur_cap -= plant
return ans第一次提交的版本
class Solution:
def wateringPlants(self, plants: List[int], capacity: int) -> int:
cur_cap = capacity
ans = 0
for i, plant in enumerate(plants):
ans += 1
if cur_cap < plant:
ans += 2 * i
cur_cap = capacity
cur_cap -= plant
return ans