Leetcode—2079. 给植物浇水【中等】

源代码•宸

于 2024-05-08 09:11:08 发布

阅读量208

点赞数 3

分类专栏： LeetCode刷题文章标签： leetcode 算法经验分享 c++

本文链接：https://blog.csdn.net/qq_44631615/article/details/138557534

版权

LeetCode刷题专栏收录该内容

378 篇文章 5 订阅

订阅专栏

2024每日刷题（130）

Leetcode—2079. 给植物浇水

在这里插入图片描述

实现代码

class Solution {
public:
    int wateringPlants(vector<int>& plants, int capacity) {
        int ans = 0;
        int step = 0;
        int cap = capacity;
        bool flag = false;
        for(int i = 0; i < plants.size(); i++) {
            if(capacity >= plants[i]) {
                step++;
                capacity -= plants[i];
                flag = true;
            } else {
                flag = false;
                int tmp = step;
                step *= 2;
                ans += step;
                step = tmp;
                i = step - 1;
                capacity = cap;
            }
            if(i == plants.size() - 1 && flag) {
                ans += step;
            }
        }
        return ans;
    }
};