发表于 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

• LC 题目链接