LeetCode系列5201—给植物浇水

最新推荐文章于 2024-05-15 00:15:00 发布

fkuner

最新推荐文章于 2024-05-15 00:15:00 发布

阅读量158

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/younothings/article/details/121451902

版权

LeetCode 专栏收录该内容

133 篇文章 0 订阅

订阅专栏

题意

5201. 给植物浇水

题解

模拟

class Solution {
public:
    int wateringPlants(vector<int>& plants, int capacity) {
        int ans = 0, now = capacity, pos = -1, n = plants.size();
        for (int i = 0; i < n; ++i) {
            ans += i - pos;
            pos = i;
            now -= plants[i];
            if (i + 1 < n && now < plants[i + 1]) {
                pos = -1;
                ans += i - pos;
                now = capacity;
            }
        }
        return ans;
    }
};

class Solution:
    def wateringPlants(self, plants: List[int], capacity: int) -> int:
        ans, cur = 0, capacity
        for i, plant in enumerate(plants):
            if cur < plant:
                cur = capacity
                ans += 2 * i + 1
            else:
                ans += 1
            cur -= plant
        return ans

func wateringPlants(plants []int, capacity int) int {
	ans := len(plants)
	water := capacity // 初始水量
	for i, need := range plants {
		if water < need {
			ans += 2 * i     // 往返
			water = capacity // 重置水量
		}
		water -= need
	}
	return ans
}