【Leetcode】【python】Candy 分发糖果

题目大意

一直线上站了N个孩子，每个孩子都有一个属于自己的数字，现在按照如下规则给孩子分发糖果：每个孩子至少有一个糖果；相邻的孩子中数字比较大的那个拿的糖果也比较多。求最少要发掉多少个糖果。

解题思路

贪心法。
想象下，先从前面开始升序遍历，所有升序的就从1开始给，再从后面开始升序遍历，所有升序的就从1开始给，遇到所有需要改变值的节点，对比想要改成的值与之前的值，取最大的保证其对两边都是最大。

注意：
在数组一开始就为降序时，如下例，他的第1和第2个孩子会先分到同样的1个。要从后面开始扫描，将降序的较高孩子分得更多的糖果。

我的输入
[1,0,2]
我的标准输出
[1, 1, 2]
[2, 1, 2]
我的答案
5

代码

class Solution(object):
    def candy(self, ratings):
        """
        :type ratings: List[int]
        :rtype: int
        """
        n = len(ratings)
        result = [1 for __ in range(n)]
        for i in range(1, n):
            if ratings[i-1] < ratings[i]:
                result[i] = result[i-1] + 1
        # 要从后面开始扫描，将降序的较高孩子分得更多的糖果。
        for j in range(n-2, -1, -1):
            if ratings[j] > ratings[j+1]:
                result[j] = max(result[j], result[j+1] + 1)
        return sum(result)

总结