2105. 给植物浇水 II

给植物浇水 II

• 难度: medium
• 原始链接: https://leetcode.cn/problems/watering-plants-ii
• 标签: 贪心

解法一: 贪心

go

func minimumRefill(plants []int, capacityA int, capacityB int) int {
  a := capacityA
  b := capacityB
  ans := 0
  left := 0
  right := len(plants) - 1
  for left <= right {
    if left == right {
      if a < plants[left] && b < plants[left] {
        ans++
      }
      break
    }
    if a < plants[left] {
      ans++
      a = capacityA
    }
    a -= plants[left]
    if b < plants[right] {
      ans++
      b = capacityB
    }
    b -= plants[right]
    left++
    right--
  }
  return ans
}