老師想給孩子們分發糖果,有 N 個孩子站成了一條直線,老師會根據每一個孩子的表現,預先給他們評分。
你須要按照如下要求,幫助老師給這些孩子分發糖果:
每一個孩子至少分配到 1 個糖果。
相鄰的孩子中,評分高的孩子必須得到更多的糖果。
那麼這樣下來,老師至少須要準備多少顆糖果呢?數組
來源:力扣(LeetCode)
連接:https://leetcode-cn.com/problems/candycode
輸入: [1,0,2]
輸出: 5
解釋: 你能夠分別給這三個孩子分發 二、一、2 顆糖果。leetcode
輸入: [1,2,2]
輸出: 4
解釋: 你能夠分別給這三個孩子分發 一、二、1 顆糖果。
第三個孩子只獲得 1 顆糖果,這已知足上述兩個條件get
兩個數組left和right分別記錄從左向右規則和從右向左規則時,每一個孩子應該分的糖果數。
題目要求即爲同時知足兩個方向的規則,即兩個數組按位取max。
最終結果要加上每一個孩子至少一個糖果。(或者left,right數組初始化爲1)io
class Solution { public int candy(int[] ratings) { int len = ratings.length; int sum = len; int[] left = new int[len]; int[] right = new int[len]; for (int i = 1; i < len; i++) { if(ratings[i] > ratings[i-1]) { left[i] = left[i-1]+1; } } for (int i = len-2; i>=0; i--) { if(ratings[i] > ratings[i+1]) { right[i] = right[i+1]+1; } } for (int i = 0; i < len; i++) { sum += Math.max(left[i], right[i]); } return sum; } }