老師想給孩子們分發糖果,有 n 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。
你需要按照以下要求,幫助老師給這些孩子分發糖果:
那麼這樣下來,老師至少需要準備多少顆糖果呢?
輸入:[1
,0,2
]輸出:
5解釋: 你可以分別給這三個孩子分發 2、1、2 顆糖果。
輸入:[1
,2,2
]輸出:
4解釋: 你可以分別給這三個孩子分發 1、2、1 顆糖果。
第三個孩子只得到 1 顆糖果,這已滿足上述兩個條件。
這個題還是比較容易想的,主要是判斷公升序還是降序,遍歷一遍就可以得到答案。
class
solution
:def
candy
(self, ratings: list[
int])-
>
int:
float
("inf"))
res =
0 num =
1 i =
0while i <
len(ratings)-1
:if ratings[i]
< ratings[i +1]
: res += num
num +=
1 i +=
1elif ratings[i]
== ratings[i +1]
: res += num
num =
1 i +=
1else
: j = i +
1while j <
len(ratings)
and ratings[j]
> ratings[j+1]
: j +=
1 seq_len = j - i
res +=(1
+ seq_len)
*seq_len//2-
1+max(num, seq_len +1)
num =
1 i = j
return res
LeetCode 135 分發糖果
老師想給孩子們分發糖果,有 n 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。你需要按照以下要求,幫助老師給這些孩子分發糖果 每個孩子至少分配到 1 個糖果。相鄰的孩子中,評分高的孩子必須獲得更多的糖果。那麼這樣下來,老師至少需要準備多少顆糖果呢?示例 1 輸入 1,0,2 輸出 ...
leetcode135 分發糖果
一道很有意思的題目,難度級別為 困難。題目是這樣的的 老師想給孩子們分發糖果,有 n 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。你需要按照以下要求,幫助老師給這些孩子分發糖果 每個孩子至少分配到 1 個糖果。相鄰的孩子中,評分高的孩子必須獲得更多的糖果。那麼這樣下來,老師至少需...
leetcode135 分發糖果
老師想給孩子們分發糖果,有 n 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。你需要按照以下要求,幫助老師給這些孩子分發糖果 每個孩子至少分配到 1 個糖果。相鄰的孩子中,評分高的孩子必須獲得更多的糖果。那麼這樣下來,老師至少需要準備多少顆糖果呢?示例 1 輸入 1,0,2 輸出 ...