1. 程式人生 > 其它 >LeetCode-135. 分發糖果(python3.7)

LeetCode-135. 分發糖果(python3.7)

技術標籤:practiceleetcodepython

老師想給孩子們分發糖果,有 N 個孩子站成了一條直線,老師會根據每個孩子的表現,預先給他們評分。

你需要按照以下要求,幫助老師給這些孩子分發糖果:

每個孩子至少分配到 1 個糖果。
相鄰的孩子中,評分高的孩子必須獲得更多的糖果。
那麼這樣下來,老師至少需要準備多少顆糖果呢?

示例 1:

輸入: [1,0,2]
輸出: 5
解釋: 你可以分別給這三個孩子分發 212 顆糖果。

示例 2:

輸入: [1,2,2]
輸出: 4
解釋: 你可以分別給這三個孩子分發 121 顆糖果。
     第三個孩子只得到 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/