5201. 給植物澆水

你打算用一個水罐給花園里的 n 株植物澆水。植物排成一行，從左到右進行標記，編號從 0 到 n - 1 。其中，第 i 株植物的位置是 x = i 。x = -1?處有一條河，你可以在那里重新灌滿你的水罐。

每一株植物都需要澆特定量的水。你將會按下面描述的方式完成澆水：

• 按從左到右的順序給植物澆水。
• 在給當前植物澆完水之后，如果你沒有足夠的水 完全 澆灌下一株植物，那么你就需要返回河邊重新裝滿水罐。
• 你 不能 提前重新灌滿水罐。
  最初，你在河邊（也就是，x = -1），在 x 軸上每移動 一個單位?都需要 一步 。

給你一個下標從 0 開始的整數數組 plants ，數組由 n 個整數組成。其中，plants[i] 為第 i 株植物需要的水量。另有一個整數 capacity 表示水罐的容量，返回澆灌所有植物需要的 步數 。

示例 1：輸入：plants = [2,2,3,3], capacity = 5
輸出：14
解釋：從河邊開始，此時水罐是裝滿的：
- 走到植物 0 (1 步) ，澆水。水罐中還有 3 單位的水。
- 走到植物 1 (1 步) ，澆水。水罐中還有 1 單位的水。
- 由于不能完全澆灌植物 2 ，回到河邊取水 (2 步)。
- 走到植物 2 (3 步) ，澆水。水罐中還有 2 單位的水。
- 由于不能完全澆灌植物 3 ，回到河邊取水 (3 步)。
- 走到植物 3 (4 步) ，澆水。
需要的步數是 = 1 + 1 + 2 + 3 + 3 + 4 = 14 。示例 2：輸入：plants = [1,1,1,4,2,3], capacity = 4
輸出：30
解釋：從河邊開始，此時水罐是裝滿的：
- 走到植物 0，1，2 (3 步) ，澆水。回到河邊取水 (3 步)。
- 走到植物 3 (4 步) ，澆水。回到河邊取水 (4 步)。
- 走到植物 4 (5 步) ，澆水。回到河邊取水 (5 步)。
- 走到植物 5 (6 步) ，澆水。
需要的步數是 = 3 + 3 + 4 + 4 + 5 + 5 + 6 = 30 。示例 3：輸入：plants = [7,7,7,7,7,7,7], capacity = 8
輸出：49
解釋：每次澆水都需要重新灌滿水罐。
需要的步數是 = 1 + 1 + 2 + 2 + 3 + 3 + 4 + 4 + 5 + 5 + 6 + 6 + 7 = 49 。

提示：

• n == plants.length
• 1 <= n <= 1000
• 1 <= plants[i] <= $10^6$
• max(plants[i]) <= capacity <= $10^9$

解題思路

當我們走到某株植物前，判斷水罐里面的水，是否夠用，如果發現了不夠水，需要返回河邊重新裝滿水罐，一來一回的步數為2*(i+1)-1，i為當前缺水植物的下標，而正常足夠水澆灌的話，我們只需要每次增加一步。

代碼

class Solution {
public:int wateringPlants(vector<int>& plants, int capacity) {int cur=capacity,res=0;for (int i = 0; i < plants.size(); ++i) {if (cur<plants[i]){res+=(2*(i+1)-1);cur=capacity-plants[i];}else {cur-=plants[i];res++;}}return res;}
};