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

最新推荐文章于 2024-09-21 10:02:17 发布

源代码•宸

最新推荐文章于 2024-09-21 10:02:17 发布

阅读量327

点赞数 5

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

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

版权

LeetCode刷题专栏收录该内容

378 篇文章 5 订阅

订阅专栏

2024每日刷题（131）

Leetcode—2105. 给植物浇水 II

在这里插入图片描述

实现代码

class Solution {
public:
    int minimumRefill(vector<int>& plants, int capacityA, int capacityB) {
        int size = plants.size();
        
        int i = 0;
        int j = size - 1;
        int capA = capacityA;
        int capB = capacityB;
        int ans = 0;
        while(i < j) {
            ans += (capA < plants[i]) + (capB < plants[j]);
            if(capA < plants[i]) {
                capA = capacityA;
            }
            if(capB < plants[j]) {
                capB = capacityB;
            }
            capA -= plants[i++];
            capB -= plants[j--];
        }
        return (i == j) ? (max(capA, capB) < plants[i]) + ans: ans;
    }
};