leetcode 2105. 给植物浇水 II

2105. 给植物浇水 II

题目描述

Alice 和 Bob 打算给花园里的 n 株植物浇水。植物排成一行，从左到右进行标记，编号从 0 到 n - 1 。每一株植物都需要浇特定量的水。Alice 和 Bob 每人有一个水罐，最初是满的。他们按照以下描述的方式完成浇水：

• Alice 按从左到右的顺序给植物浇水，从植物 0 开始。
• Bob 按从右到左的顺序给植物浇水，从植物 n - 1 开始。
• 他们同时给植物浇水。
• 如果没有足够的水完全浇灌下一株植物，他/她会立即重新灌满浇水罐。
• 每株植物都可以由 Alice 或者 Bob 来浇水。
• 如果 Alice 和 Bob 到达同一株植物，那么当前水罐中水更多的人会给这株植物浇水。如果他俩水量相同，那么 Alice 会给这株植物浇水。

给你一个下标从 0 开始的整数数组 plants，数组由 n 个整数组成。其中，plants[i] 为第 i 株植物需要的水量。另有两个整数 capacityA 和 capacityB 分别表示 Alice 和 Bob 水罐的容量。返回两人浇灌所有植物过程中重新灌满水罐的次数。

示例

示例 1

输入：plants = [2,2,3,3], capacityA = 5, capacityB = 5
输出：1
解释：
- 最初，Alice 和 Bob 的水罐中各有 5 单元水。
- Alice 给植物 0 浇水，Bob 给植物 3 浇水。
- Alice 和 Bob 现在分别剩下 3 单元和 2 单元水。
- Alice 有足够的水给植物 1 ，所以她直接浇水。Bob 的水不够给植物 2 ，所以他先重新装满水，再浇水。
所以，两人浇灌所有植物过程中重新灌满水罐的次数 = 0 + 0 + 1 + 0 = 1 。

示例 2

输入：plants = [2,2,3,3], capacityA = 3, capacityB = 4
输出：2
解释：
- 最初，Alice 的水罐中有 3 单元水，Bob 的水罐中有 4 单元水。
- Alice 给植物 0 浇水，Bob 给植物 3 浇水。
- Alice 和 Bob 现在都只有 1 单元水，并分别需要给植物 1 和植物 2 浇水。
- 由于他们的水量均不足以浇水，所以他们重新灌满水罐再进行浇水。
所以，两人浇灌所有植物过程中重新灌满水罐的次数 = 0 + 1 + 1 + 0 = 2 。

示例 3

输入：plants = [5], capacityA = 10, capacityB = 8
输出：0
解释：
- 只有一株植物
- Alice 的水罐有 10 单元水，Bob 的水罐有 8 单元水。因此 Alice 的水罐中水更多，她会给这株植物浇水。
所以，两人浇灌所有植物过程中重新灌满水罐的次数 = 0 。

分析

这道题目的目标是模拟 Alice 和 Bob 同时给植物浇水的过程，并计算他们重新灌满水罐的次数。题目中给出了植物的需水量数组 plants，以及 Alice 和 Bob 的水罐容量 capacityA 和 capacityB。他们从两端向中间浇水，同时比较谁的水更多来决定谁给当前植物浇水。

具体实现的思路是使用下标，indexA 从左向右遍历，indexB 从右向左遍历。因为每次浇水时间是相同的，因此每次循环只需要进行index++和–就行，根据两个index所示植物的需水量和当前水罐的水量，来决定是否需要重新灌满水罐，并统计重新灌满的次数。

实现

#include "stdio.h"

// 定义一个宏，用于取两个数中的较大值
#define MAX(a,b) (a > b ? a : b)

// 计算重新灌满水罐的次数
int minimumRefill(int* plants, int plantsSize, int capacityA, int capacityB) {
    // 初始化 Alice 和 Bob 的水罐当前水量
    int nowWtA = capacityA; // Alice 的水罐当前水量
    int nowWtB = capacityB; // Bob 的水罐当前水量
    int reloadTimes = 0;    // 重新灌满水罐的次数
    int indexA = 0, indexB = plantsSize - 1; // 左右下标，indexA 从左向右遍历，indexB 从右向左遍历
    
    // 使用 do-while 循环遍历植物数组，直到两个下标相遇
    do {
        // 如果左右指针指向同一个植物
        if (indexA == indexB) {
            // 判断当前植物的需水量是否大于两个水罐当前水量的最大值
            if (plants[indexA] > MAX(nowWtA, nowWtB)) {
                reloadTimes++; // 需要重新灌满水罐
            }
            break; // 跳出循环
        }
        
        // 处理左下标指向的植物（alice）
        if (plants[indexA] > nowWtA) { // 如果当前植物需要的水量大于 Alice 的水罐当前水量
            reloadTimes++;              // Alice 需要重新灌满水罐
            nowWtA = capacityA - plants[indexA]; // Alice 重新装满水罐并浇水给当前植物
        } else {
            nowWtA -= plants[indexA]; // Alice 直接从水罐中浇水给当前植物
        }
        
        // 处理右下标指向的植物（bob）
        if (plants[indexB] > nowWtB) { // 如果当前植物需要的水量大于 Bob 的水罐当前水量
            reloadTimes++;              // Bob 需要重新灌满水罐
            nowWtB = capacityB - plants[indexB]; // Bob 重新装满水罐并浇水给当前植物
        } else {
            nowWtB -= plants[indexB]; // Bob 直接从水罐中浇水给当前植物
        }
        
        indexA++; // 左下标向右移动
        indexB--; // 右下标向左移动
    } while (indexA <= indexB); // 当两个下标相同或交错时停止循环
     
    return reloadTimes; // 返回重新灌满水罐的总次数
}

// 主函数
int main(int argc, void* argv) {
    // 定义植物数组和水罐容量
    int plant[] = {2, 2, 3, 3}; // 植物需要的水量数组
    int capA = 5, capB = 5;     // Alice 和 Bob 的水罐容量
    int plantsSize = sizeof(plant) / sizeof(plant[0]); // 计算植物数组的长度
    // 调用函数计算重新灌满水罐的次数
    int ret = minimumRefill(plant, plantsSize, capA, capB);
    // 输出结果
    printf("ret = %d", ret);
}

### 结果
非常简单
![结果图](https://img-blog.csdnimg.cn/direct/39151b1409ea447e8b41072c284c47de.png)