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 * isum += 2 * i;// 重新装满水罐water = capacity;}// 浇水water -= plant;// 步数 +1sum++;}return sum;}
};
// @lc code=end
结果:
复杂度分析:
时间复杂度:O(n),其中 n 是数组 plants 的长度。
空间复杂度:O(1)。