class Solution {
public:
int minimumRefill(vector<int>& plants, int capacityA, int capacityB) {
}
};
2105. 给植物浇水 II
Alice 和 Bob 打算给花园里的 n 株植物浇水。植物排成一行，从左到右进行标记，编号从 0 到 n - 1 。其中，第 i 株植物的位置是 x = i 。
每一株植物都需要浇特定量的水。Alice 和 Bob 每人有一个水罐，最初是满的 。他们按下面描述的方式完成浇水：
0 开始。Bob 按 从右到左 的顺序给植物浇水，从植物 n - 1 开始。他们 同时 给植物浇水。给你一个下标从 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 。
提示：
n == plants.length1 <= n <= 1051 <= plants[i] <= 106max(plants[i]) <= capacityA, capacityB <= 109原站题解
rust 解法, 执行用时: 5 ms, 内存消耗: 3.5 MB, 提交时间: 2024-05-09 09:28:51
impl Solution {
pub fn minimum_refill(plants: Vec<i32>, capacity_a: i32, capacity_b: i32) -> i32 {
let mut ans = 0;
let mut a = capacity_a;
let mut b = capacity_b;
let mut i = 0;
let mut j = plants.len() - 1;
while i < j {
// Alice 给植物 i 浇水
if a < plants[i] {
// 没有足够的水，重新灌满水罐
ans += 1;
a = capacity_a;
}
a -= plants[i];
i += 1;
// Bob 给植物 j 浇水
if b < plants[j] {
// 没有足够的水，重新灌满水罐
ans += 1;
b = capacity_b;
}
b -= plants[j];
j -= 1;
}
// Alice 和 Bob 到达同一株植物，那么当前水罐中水更多的人会给这株植物浇水
if i == j && a.max(b) < plants[i] {
// 没有足够的水，重新灌满水罐
ans += 1;
}
ans
}
}
javascript 解法, 执行用时: 87 ms, 内存消耗: 60.7 MB, 提交时间: 2024-05-09 09:28:39
/**
* @param {number[]} plants
* @param {number} capacityA
* @param {number} capacityB
* @return {number}
*/
var minimumRefill = function(plants, capacityA, capacityB) {
let ans = 0;
let a = capacityA, b = capacityB;
let i = 0, j = plants.length - 1;
while (i < j) {
// Alice 给植物 i 浇水
if (a < plants[i]) {
// 没有足够的水，重新灌满水罐
ans++;
a = capacityA;
}
a -= plants[i++];
// Bob 给植物 j 浇水
if (b < plants[j]) {
// 没有足够的水，重新灌满水罐
ans++;
b = capacityB;
}
b -= plants[j--];
}
// Alice 和 Bob 到达同一株植物，那么当前水罐中水更多的人会给这株植物浇水
if (i === j && Math.max(a, b) < plants[i]) {
// 没有足够的水，重新灌满水罐
ans++;
}
return ans;
};
golang 解法, 执行用时: 90 ms, 内存消耗: 8.2 MB, 提交时间: 2024-05-09 09:28:27
func minimumRefill(plants []int, capacityA, capacityB int) (ans int) {
a, b := capacityA, capacityB
i, j := 0, len(plants)-1
for i < j {
// Alice 给植物 i 浇水
if a < plants[i] {
// 没有足够的水，重新灌满水罐
ans++
a = capacityA
}
a -= plants[i]
i++
// Bob 给植物 j 浇水
if b < plants[j] {
// 没有足够的水，重新灌满水罐
ans++
b = capacityB
}
b -= plants[j]
j--
}
// Alice 和 Bob 到达同一株植物，那么当前水罐中水更多的人会给这株植物浇水
if i == j && max(a, b) < plants[i] {
// 没有足够的水，重新灌满水罐
ans++
}
return
}
java 解法, 执行用时: 4 ms, 内存消耗: 56.7 MB, 提交时间: 2024-05-09 09:28:16
class Solution {
public int minimumRefill(int[] plants, int capacityA, int capacityB) {
int ans = 0;
int a = capacityA;
int b = capacityB;
int i = 0;
int j = plants.length - 1;
while (i < j) {
// Alice 给植物 i 浇水
if (a < plants[i]) {
// 没有足够的水，重新灌满水罐
ans++;
a = capacityA;
}
a -= plants[i++];
// Bob 给植物 j 浇水
if (b < plants[j]) {
// 没有足够的水，重新灌满水罐
ans++;
b = capacityB;
}
b -= plants[j--];
}
// Alice 和 Bob 到达同一株植物，那么当前水罐中水更多的人会给这株植物浇水
if (i == j && Math.max(a, b) < plants[i]) {
// 没有足够的水，重新灌满水罐
ans++;
}
return ans;
}
}
cpp 解法, 执行用时: 109 ms, 内存消耗: 78.8 MB, 提交时间: 2024-05-09 09:28:05
class Solution {
public:
int minimumRefill(vector<int>& plants, int capacityA, int capacityB) {
int ans = 0;
int a = capacityA, b = capacityB;
int i = 0, j = plants.size() - 1;
while (i < j) {
// Alice 给植物 i 浇水
if (a < plants[i]) {
// 没有足够的水，重新灌满水罐
ans++;
a = capacityA;
}
a -= plants[i++];
// Bob 给植物 j 浇水
if (b < plants[j]) {
// 没有足够的水，重新灌满水罐
ans++;
b = capacityB;
}
b -= plants[j--];
}
// Alice 和 Bob 到达同一株植物，那么当前水罐中水更多的人会给这株植物浇水
if (i == j && max(a, b) < plants[i]) {
// 没有足够的水，重新灌满水罐
ans++;
}
return ans;
}
};
python3 解法, 执行用时: 86 ms, 内存消耗: 31.2 MB, 提交时间: 2024-05-09 09:27:53
class Solution:
def minimumRefill(self, plants: List[int], capacityA: int, capacityB: int) -> int:
ans = 0
a, b = capacityA, capacityB
i, j = 0, len(plants) - 1
while i < j:
# Alice 给植物 i 浇水
if a < plants[i]:
# 没有足够的水，重新灌满水罐
ans += 1
a = capacityA
a -= plants[i]
i += 1
# Bob 给植物 j 浇水
if b < plants[j]:
# 没有足够的水，重新灌满水罐
ans += 1
b = capacityB
b -= plants[j]
j -= 1
# Alice 和 Bob 到达同一株植物，那么当前水罐中水更多的人会给这株植物浇水
if i == j and max(a, b) < plants[i]:
# 没有足够的水，重新灌满水罐
ans += 1
return ans
python3 解法, 执行用时: 120 ms, 内存消耗: 25.6 MB, 提交时间: 2022-08-10 14:46:15
class Solution:
def minimumRefill(self, plants: List[int], capacityA: int, capacityB: int) -> int:
res = 0 # 灌满水罐次数
n = len(plants)
posa, posb = 0, n - 1 # 两人位置
vala, valb = capacityA, capacityB # 两人剩余水量
# 模拟相遇前的浇水过程
while posa < posb:
if vala < plants[posa]:
res += 1
vala = capacityA - plants[posa]
else:
vala -= plants[posa]
posa += 1
if valb < plants[posb]:
res += 1
valb = capacityB - plants[posb]
else:
valb -= plants[posb]
posb -= 1
# 模拟相遇后可能的浇水过程
if posa == posb:
if vala >= valb and vala < plants[posa]:
res += 1
elif vala < valb and valb < plants[posb]:
res += 1
return res