问题描述
Alice 和 Bob 打算给花园里的
n
株植物浇水。植物排成一行,从左到右进行标记,编号从0
到n - 1
。其中,第i
株植物的位置是x = i
。每一株植物都需要浇特定量的水。Alice 和 Bob 每人有一个水罐,最初是满的 。他们按下面描述的方式完成浇水:
- Alice 按 从左到右 的顺序给植物浇水,从植物
0
开始。Bob 按 从右到左 的顺序给植物浇水,从植物n - 1
开始。他们 同时 给植物浇水。- 无论需要多少水,为每株植物浇水所需的时间都是相同的。
- 如果 Alice/Bob 水罐中的水足以 完全 灌溉植物,他们 必须 给植物浇水。否则,他们 首先(立即)重新装满罐子,然后给植物浇水。
- 如果 Alice 和 Bob 到达同一株植物,那么当前水罐中水 更多 的人会给这株植物浇水。如果他俩水量相同,那么 Alice 会给这株植物浇水。
给你一个下标从 0 开始的整数数组
plants
,数组由n
个整数组成。其中,plants[i]
为第i
株植物需要的水量。另有两个整数capacityA
和capacityB
分别表示 Alice 和 Bob 水罐的容量。返回两人浇灌所有植物过程中重新灌满水罐的 次数 。
力扣官方题解
class Solution {
public int minimumRefill(int[] plants, int capacityA, int capacityB) {
int res = 0; // 灌满水罐次数
int n = plants.length;
int posa = 0, posb = n - 1; // 两人位置
int vala = capacityA, valb = capacityB; // 两人剩余水量
// 模拟相遇前的浇水过程
while (posa < posb) {
if (vala < plants[posa]) {
++res;
vala = capacityA - plants[posa];
} else {
vala -= plants[posa];
}
++posa;
if (valb < plants[posb]) {
++res;
valb = capacityB - plants[posb];
} else {
valb -= plants[posb];
}
--posb;
}
// 模拟相遇后可能的浇水过程
if (posa == posb) {
if (vala >= valb && vala < plants[posa]) {
++res;
}
if (vala < valb && valb < plants[posb]) {
++res;
}
}
return res;
}
}