2228. watering plants ii

Author: eirture

README.md

@@ -19,6 +19,7 @@ make commit
 
 | # | Title | Difficulty | Date |
 |---| ----- | ---------- | ---- |
+|2228|[给植物浇水 II](./algorithms/2228-watering-plants-ii.md)|medium|2024/05/09|
 |11|[盛最多水的容器](./algorithms/11-container-with-most-water.md)|medium|2023/11/08|
 |2654|[统计范围内的元音字符串数](./algorithms/2654-count-the-number-of-vowel-strings-in-range.md)|easy|2023/11/07|
 |307|[区域和检索 - 数组可修改](./algorithms/307-range-sum-query-mutable.md)|medium|2022/04/05|

algorithms/2228-watering-plants-ii.md

@@ -0,0 +1,97 @@
+# [2228. 给植物浇水 II](https://leetcode.cn/problems/watering-plants-ii/) - medium
+
+<p>Alice 和 Bob 打算给花园里的 <code>n</code> 株植物浇水。植物排成一行，从左到右进行标记，编号从 <code>0</code> 到 <code>n - 1</code> 。其中，第 <code>i</code> 株植物的位置是 <code>x = i</code> 。</p>
+
+<p>每一株植物都需要浇特定量的水。Alice 和 Bob 每人有一个水罐，<strong>最初是满的 </strong>。他们按下面描述的方式完成浇水：</p>
+
+<ul>
+	<li>&nbsp;Alice 按 <strong>从左到右</strong> 的顺序给植物浇水，从植物 <code>0</code> 开始。Bob 按 <strong>从右到左</strong> 的顺序给植物浇水，从植物 <code>n - 1</code> 开始。他们 <strong>同时</strong> 给植物浇水。</li>
+	<li>无论需要多少水，为每株植物浇水所需的时间都是相同的。</li>
+	<li>如果 Alice/Bob 水罐中的水足以 <strong>完全</strong> 灌溉植物，他们 <strong>必须</strong> 给植物浇水。否则，他们 <strong>首先</strong>（立即）重新装满罐子，然后给植物浇水。</li>
+	<li>如果 Alice 和 Bob 到达同一株植物，那么当前水罐中水 <strong>更多</strong> 的人会给这株植物浇水。如果他俩水量相同，那么 Alice 会给这株植物浇水。</li>
+</ul>
+
+<p>给你一个下标从 <strong>0</strong> 开始的整数数组 <code>plants</code> ，数组由 <code>n</code> 个整数组成。其中，<code>plants[i]</code> 为第 <code>i</code> 株植物需要的水量。另有两个整数 <code>capacityA</code> 和&nbsp;<code>capacityB</code> 分别表示 Alice 和 Bob 水罐的容量。返回两人浇灌所有植物过程中重新灌满水罐的 <strong>次数</strong> 。</p>
+
+<p>&nbsp;</p>
+
+<p><strong>示例 1：</strong></p>
+
+<pre>
+<strong>输入：</strong>plants = [2,2,3,3], capacityA = 5, capacityB = 5
+<strong>输出：</strong>1
+<strong>解释：</strong>
+- 最初，Alice 和 Bob 的水罐中各有 5 单元水。
+- Alice 给植物 0 浇水，Bob 给植物 3 浇水。
+- Alice 和 Bob 现在分别剩下 3 单元和 2 单元水。
+- Alice 有足够的水给植物 1 ，所以她直接浇水。Bob 的水不够给植物 2 ，所以他先重新装满水，再浇水。
+所以，两人浇灌所有植物过程中重新灌满水罐的次数 = 0 + 0 + 1 + 0 = 1 。</pre>
+
+<p><strong>示例 2：</strong></p>
+
+<pre>
+<strong>输入：</strong>plants = [2,2,3,3], capacityA = 3, capacityB = 4
+<strong>输出：</strong>2
+<strong>解释：</strong>
+- 最初，Alice 的水罐中有 3 单元水，Bob 的水罐中有 4 单元水。
+- Alice 给植物 0 浇水，Bob 给植物 3 浇水。
+- Alice 和 Bob 现在都只有 1 单元水，并分别需要给植物 1 和植物 2 浇水。
+- 由于他们的水量均不足以浇水，所以他们重新灌满水罐再进行浇水。
+所以，两人浇灌所有植物过程中重新灌满水罐的次数 = 0 + 1 + 1 + 0 = 2 。</pre>
+
+<p><strong>示例 3：</strong></p>
+
+<pre>
+<strong>输入：</strong>plants = [5], capacityA = 10, capacityB = 8
+<strong>输出：</strong>0
+<strong>解释：</strong>
+- 只有一株植物
+- Alice 的水罐有 10 单元水，Bob 的水罐有 8 单元水。因此 Alice 的水罐中水更多，她会给这株植物浇水。
+所以，两人浇灌所有植物过程中重新灌满水罐的次数 = 0 。</pre>
+
+<p>&nbsp;</p>
+
+<p><strong>提示：</strong></p>
+
+<ul>
+	<li><code>n == plants.length</code></li>
+	<li><code>1 &lt;= n &lt;= 10<sup>5</sup></code></li>
+	<li><code>1 &lt;= plants[i] &lt;= 10<sup>6</sup></code></li>
+	<li><code>max(plants[i]) &lt;= capacityA, capacityB &lt;= 10<sup>9</sup></code></li>
+</ul>
+
+
+## Solutions
+
+### 1. 双指针遍历
+
+一个简单的思路，两个指针 a，b，左右每次各往中间走一步。每次判断 va，vb 记录 Alice 和 Bob 水壶中的水。比较植物需要的水，如果需要加水则计数 +1。直到 a == b。
+
+时间复杂度是 O(n)；空间复杂度是 O(1)
+
+```go
+func minimumRefill(plants []int, capacityA int, capacityB int) int {
+    a, b := 0, len(plants) - 1
+    va, vb := capacityA, capacityB
+    var n int
+    for a < b {
+        if va < plants[a] {
+            va = capacityA
+            n++
+        }
+        va -= plants[a]
+        a++
+        
+        if vb < plants[b] {
+            vb = capacityB
+            n++
+        }
+        vb -= plants[b]
+        b--
+    }
+    if a == b && va < plants[a] && vb < plants[b] {
+        n++
+    }
+    return n
+}
+```