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);
}### 结果
非常简单
