Leetcode2079. 给植物浇水

Every day a Leetcode

题目来源：2079. 给植物浇水

解法1：模拟

sum 记录总步数，初始化为 0。water 表示当前水量，初始化为 capacity。

遍历 plants 数组，设下标为 i 的植物需要的水量 plant = plants[i]。

如果 water < plant，说明没有足够的水完全浇灌该植物，需要回到河边取水，来回的步数为 2 * i，加到 sum 里，把水加满，water = capacity。

浇水，water -= plant，步数 +1。

返回 sum 即为答案。

代码：

/*
 * @lc app=leetcode.cn id=2079 lang=cpp
 *
 * [2079] 给植物浇水
 */

// @lc code=start
class Solution
{
public:
    int wateringPlants(vector<int> &plants, int capacity)
    {
        int n = plants.size();
        int sum = 0;
        int water = capacity;
        for (int i = 0; i < n; i++)
        {
            int plant = plants[i];
            // 没有足够的水完全浇灌下一株植物
            if (water < plant)
            {
                // 回到河边取水，来回的步数为 2 * i
                sum += 2 * i;
                // 重新装满水罐
                water = capacity;
            }
            // 浇水
            water -= plant;
            // 步数 +1
            sum++;
        }
        return sum;
    }
};
// @lc code=end

结果：

复杂度分析：

时间复杂度：O(n)，其中 n 是数组 plants 的长度。

空间复杂度：O(1)。