LeetCode-135. 分發糖果(python3.7)
阿新 • • 發佈:2020-12-25
老師想給孩子們分發糖果,有 N 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。
你需要按照以下要求,幫助老師給這些孩子分發糖果:
每個孩子至少分配到 1 個糖果。
相鄰的孩子中,評分高的孩子必須獲得更多的糖果。
那麼這樣下來,老師至少需要準備多少顆糖果呢?
示例 1:
輸入: [1,0,2]
輸出: 5
解釋: 你可以分別給這三個孩子分發 2、1、2 顆糖果。
示例 2:
輸入: [1,2,2] 輸出: 4 解釋: 你可以分別給這三個孩子分發 1、2、1 顆糖果。 第三個孩子只得到 1 顆糖果,這已滿足上述兩個條件。
沒想到,轉官方程式碼如下
思路:
我們可以將「相鄰的孩子中,評分高的孩子必須獲得更多的糖果」這句話拆分為兩個規則,分別處理。
我們遍歷該陣列兩次,處理出每一個學生分別滿足左規則或右規則時,最少需要被分得的糖果數量。每個人最終分得的糖果數量即為這兩個數量的最大值。
class Solution:
def candy(self, ratings: List[int]) -> int:
n = len(ratings)
left = [0] * n
for i in range(n):
if i > 0 and ratings[i] > ratings[i - 1]:
left[i] = left[i - 1] + 1
else:
left[i] = 1
right = ret = 0
for i in range(n - 1, -1, -1):
if i < n - 1 and ratings[i] > ratings[i + 1]:
right += 1
else:
right = 1
ret += max(left[i], right)
return ret
作者:LeetCode-Solution
連結:https://leetcode-cn.com/problems/candy/solution/fen-fa-tang-guo-by-leetcode-solution-f01p/