2079-watering-plants.js

/**
 * 2079. Watering Plants
 * https://leetcode.com/problems/watering-plants/
 * Difficulty: Medium
 *
 * You want to water n plants in your garden with a watering can. The plants are arranged in
 * a row and are labeled from 0 to n - 1 from left to right where the ith plant is located at
 * x = i. There is a river at x = -1 that you can refill your watering can at.
 *
 * Each plant needs a specific amount of water. You will water the plants in the following way:
 * - Water the plants in order from left to right.
 * - After watering the current plant, if you do not have enough water to completely water the
 *   next plant, return to the river to fully refill the watering can.
 * - You cannot refill the watering can early.
 *
 * You are initially at the river (i.e., x = -1). It takes one step to move one unit on the x-axis.
 *
 * Given a 0-indexed integer array plants of n integers, where plants[i] is the amount of water
 * the ith plant needs, and an integer capacity representing the watering can capacity, return
 * the number of steps needed to water all the plants.
 */

/**
 * @param {number[]} plants
 * @param {number} capacity
 * @return {number}
 */
var wateringPlants = function(plants, capacity) {
  let water = capacity;
  let result = 0;

  for (let i = 0; i < plants.length; i++) {
    if (water < plants[i]) {
      result += 2 * i;
      water = capacity;
    }
    water -= plants[i];
    result++;
  }

  return result;
};