【贪心】leetcode135.分发糖果

题目：
老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。你需要按照以下要求，帮助老师给这些孩子分发糖果：
每个孩子至少分配到 1 个糖果。
评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。
那么这样下来，老师至少需要准备多少颗糖果呢？

思路：
将每个数当看作一个波浪线中的峰值，从左向右扫描一次，再从右向左扫描一次，扫描到当前数即停止，取两者中较大的值。具体过程如下：
扫描两次：
第一次，从左向右扫描，将第一个人拥有的糖果数设定为1，即left[0]=1。扫描时，若当前值大于前一个值，则将当前值在前一个值的基础上加1，否则令当前值为1。
第二次，从右向左扫描，边扫描边判断，并对最终结果进行累加。将第一个人拥有的糖果数设定为1，即right[n-1]=0。扫描时，若当前值大于前一个值，则将当前值在前一个值的基础上加1，否则令当前值为1。取left[i]和right[i]中较大的数，对最终结果进行累加。

解答：

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

方法二：仅用一个数组

class Solution:
    def candy(self, ratings: List[int]) -> int:
        n=len(ratings)
        res=[1]*n
        #一次是从左到右遍历，只比较右边孩子评分比左边大的情况。
        for i in range(1,n):
            if ratings[i]>ratings[i-1]:
                res[i]=res[i-1]+1
        ans=res[-1]
        #一次是从右到左遍历，只比较左边孩子评分比右边大的情况。
        for i in range(n-2,-1,-1):
            if ratings[i]>ratings[i+1]:
                res[i]=max(res[i],res[i+1]+1)
            ans+=res[i]
        return ans