leetcode135 分發糖果

2021-09-18 03:10:38 字數 2704 閱讀 7395

一道很有意思的題目,難度級別為 困難。

題目是這樣的的

老師想給孩子們分發糖果,有 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 輸出 ...