一道很有意思的題目,難度級別為 困難。
題目是這樣的的
老師想給孩子們分發糖果,有 n 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。
你需要按照以下要求,幫助老師給這些孩子分發糖果:
每個孩子至少分配到 1 個糖果。
相鄰的孩子中,評分高的孩子必須獲得更多的糖果。
那麼這樣下來,老師至少需要準備多少顆糖果呢?
示例 1:
輸入: [1,0,2]
輸出: 5
解釋: 你可以分別給這三個孩子分發 2、1、2 顆糖果。
示例 2:
輸入: [1,2,2]
輸出: 4
解釋: 你可以分別給這三個孩子分發 1、2、1 顆糖果。
第三個孩子只得到 1 顆糖果,這已滿足上述兩個條件。
這題用貪心做,聽說也可以用dp。
有一種策略是先每個小朋友發一顆糖,然後從左右各遍歷一次補給評分高的同學,這樣是可以的。
但是我借鑑了大神的更有趣的做法。
先舉個例子。ratings = [1,3,5,7,6,5,4,3,2,1],下面講一下思路。
首先我們從頭開始,如果一路公升序,比如這裡的1,3,5,7,那麼我們貪心的做,自然從第乙個人給第一塊糖開始,,每人多加一塊,直到遇到了降序,注意,這裡的降序很重要!因為我們需要盡可能的少給糖,但是需要避免的是,到最後一人無糖可發。
這個例子裡面,到7這個小朋友的時候,我們給他4塊糖,那麼到6這個同學的時候,我們就給3塊糖,然後依次遞減,這時候我們會發現什麼?到了3這個小朋友的時候,我們已經無糖可給,而如果給3小朋友1塊或兩塊糖卻又違背題目規則了。
題目有趣的地方就在這,那麼我們該怎麼做?
貪心策略就是先得到降序的位數,然後我們用等差求和,公差為1(畢竟我們是貪心的),算出遞減這個階段要給的糖的個數,在這個例子裡,我們給最後乙個小朋友 1塊糖。因為他就是遞減的末尾。不管遞減末尾小朋友的評分為多少,他只有1塊糖,這是本題的貪心策略。
重點來了,就是補糖,我們不可以讓評分為7的小朋友拿到的糖比6少,但是7前面的5比6少就無所謂了,畢竟題目只要求相鄰,例子裡7 有4塊糖,而6在從1一路加糖開始,已經有了6塊,那7小朋友肯定就不樂意了,所以要補糖,補上的就是6-4+1 = 3顆,這時7就是7顆,比6多剛好多一顆。完美而又貪心。
這裡的求和公式就是求1~n的和: sum = (1 + n) * n / 2.
還需要注意的是結尾的邊界情況。這題就存在邊界情況。
然後上**。
這裡貼出c和python3的。
其中python3在leetcode上測試通過的。
#include#define length(a) ((sizeof(a))/sizeof(a[0]))
int candy(int* ratings, int ratingssize)
if(ratings[i] == ratings[i-1])
pre = 1;
else //在條件ratings[i] >= ratings[i-1]之下,如果不等,即ratings[i] > ratings[i-1],此時只需要順序加糖果
pre += 1;
res += pre;
} else //倒序的時候
dis += 1;
} if(dis > 0)
}return (int)res;
}int main()
; int ratingssize = length(ratings);
int ans = candy(ratings,ratingssize);
printf("%d",ans);
return 0;
}
class solution:
def candy(self, ratings) :
if not ratings:
return 0
res , pre , dis = 1 , 1 , 0 #dis 為保險的步長,避免後面的小朋友無糖可給,dis為至少要給多少才可保證有糖可以給
for i in range(1,len(ratings)):
if ratings[i] >= ratings[i-1]:
if dis > 0:
res += dis * (dis + 1) / 2
if dis >= pre:#補給之前小朋友的糖
res += dis - pre + 1
dis = 0
pre = 1
pre = 1 if ratings[i] == ratings[i - 1] else pre + 1
res += pre
else:
dis += 1
if dis > 0 : #最後的邊界情況
LeetCode 135 分發糖果
老師想給孩子們分發糖果,有 n 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。你需要按照以下要求,幫助老師給這些孩子分發糖果 每個孩子至少分配到 1 個糖果。相鄰的孩子中,評分高的孩子必須獲得更多的糖果。那麼這樣下來,老師至少需要準備多少顆糖果呢?示例 1 輸入 1,0,2 輸出 ...
leetcode135 分發糖果
老師想給孩子們分發糖果,有 n 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。你需要按照以下要求,幫助老師給這些孩子分發糖果 每個孩子至少分配到 1 個糖果。相鄰的孩子中,評分高的孩子必須獲得更多的糖果。那麼這樣下來,老師至少需要準備多少顆糖果呢?示例 1 輸入 1,0,2 輸出 ...
leetcode135 分發糖果
老師想給孩子們分發糖果,有 n 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。你需要按照以下要求,幫助老師給這些孩子分發糖果 每個孩子至少分配到 1 個糖果。相鄰的孩子中,評分高的孩子必須獲得更多的糖果。那麼這樣下來,老師至少需要準備多少顆糖果呢?示例 1 輸入 1,0,2 輸出 ...