python Solution LeetCode No.135 Candy（分发糖果）

题目:

老师想给孩子们分发糖果，有 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:
        candyList = len(ratings)*[1]
        for i in range(len(ratings)-1):
            if ratings[i+1] > ratings[i]:
                candyList[i+1] = candyList[i] + 1
        for j in range(len(ratings)-1, 0, -1):
            if ratings[j] < ratings[j-1] and candyList[j-1] <= candyList[j]:
                candyList[j-1] = candyList[j] + 1
        return sum(candyList)

---

说明:

参考官方题解，使用正反两次循环遍历解决。首先初始化一个列表，长度等于ratings，所有元素初始值置1，保证每人都有一个糖果（这样省得以后再一个个append了）。首先从左向右逐个遍历ratings，保证左到右分数升序的小孩拿到正确的糖果数。然后从右向左遍历ratings，保证从右向左分数升序的小孩拿到正确的糖果。最后从整体来看，ratings列表无论是升序部分，还是降序部分，都得到了正确的处理，保证分发正确的糖果数。其中第二次逆序遍历时的临时值判断条件candyList[j-1] <= candyList[j]是很关键的，如果不加此判断条件，逆向遍历会把，正向遍历升序部分最大值覆盖掉。

举个例子：ratings = [1,2,3,4,5,6,0,23]

从左到右正向遍历后的 candyList = [1,2,3,4,5,6,1,2]，当从右到左进行逆序遍历时，当ratings[j]=0 时，ratings[j-1]=6，满足判断条件会将candyList中的6修改为2，得到错误结果。通过candyList[j-1] <= candyList[j]条件限制后，实际就是在取两次遍历后分别到达临界时candyList中较大者。