135. 分发糖果(贪心算法)
老師想給孩子們分發糖果,有 N 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。
你需要按照以下要求,幫助老師給這些孩子分發糖果:
每個孩子至少分配到 1 個糖果。
評分更高的孩子必須比他兩側的鄰位孩子獲得更多的糖果。
那么這樣下來,老師至少需要準備多少顆糖果呢?
示例 1:
輸入:[1,0,2]
輸出:5
解釋:你可以分別給這三個孩子分發 2、1、2 顆糖果。
示例 2:
輸入:[1,2,2]
輸出:4
解釋:你可以分別給這三個孩子分發 1、2、1 顆糖果。
第三個孩子只得到 1 顆糖果,這已滿足上述兩個條件。
————————————————————————————————————————————————————
這道題看起來思路很簡單,但是在這里需要注意一點,就是一定是要確定一邊之后,再確定另一邊,例如比較每一個孩子的左邊,然后再比較右邊,如果同時進行就會出錯;
右邊評分大于左邊的情況
局部最優:只要右邊評分比左邊大,右邊的孩子就多一個糖果,
全局最優:相鄰的孩子中,評分高的右孩子獲得比左邊孩子更多的糖果
左邊評分大于右邊的情況
這里先注意一點,就是要從后往前遍歷,如果從前向后遍歷,根據 ratings[i + 1] 來確定 ratings[i] 對應的糖果,那么每次都不能利用上前一次的比較結果了。
局部最優:取candyVec[i + 1] + 1 和 candyVec[i] 最大的糖果數量。
全局最優:相鄰的孩子中,評分高的孩子獲得更多的糖果。
代碼如下:
class Solution { public:int candy(vector<int>& ratings) {//candyVec存放每一個孩子的得到的糖果數,初始默認每個孩子的糖果數為1;vector<int> candyVec(ratings.size(), 1);//右邊評分大于左邊的情況for (int i = 1; i < ratings.size(); ++i) {if (ratings[i] > ratings[i - 1])candyVec[i] = candyVec[i - 1] + 1;}//左邊評分大于右邊的情況for (int i = ratings.size() - 2; i >= 0; --i) {if (ratings[i] > ratings[i + 1]) candyVec[i] = max(candyVec[i + 1] + 1, candyVec[i]);}//計算總和int total = 0;for (int i : candyVec) total += i;return total;} };總結
以上是生活随笔為你收集整理的135. 分发糖果(贪心算法)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 134. 加油站(贪心算法)
- 下一篇: 哈希表(散列表)知识点概述