题目大意

一直线上站了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)

总结