135.分发糖果

135.分发糖果
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求，给这些孩子分发糖果：
每个孩子至少分配到 1 个糖果。
相邻两个孩子评分更高的孩子会获得更多的糖果。
请你给每个孩子分发糖果，计算并返回需要准备的 最少糖果数目 。

示例 1：
输入：ratings = [1,0,2]
输出：5
解释：你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。

示例 2：
输入：ratings = [1,2,2]
输出：4
解释：你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果，这满足题面中的两个条件。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/candy

class Solution(object):
    def candy(self, ratings):
        """
        :type ratings: List[int]
        :rtype: int
        """
        lens = len(ratings)
        candy = 0
        result = [1] * lens
        # 正向遍历，右边的小孩比左边的多
        for i in range(lens-1):
            if ratings[i+1] > ratings[i]:
                result[i+1] = max(result[i], result[i+1]) + 1
        # 反向遍历，左边的小孩比右边的多
        for j in range(lens-1):
            if ratings[lens-j-2] > ratings[lens-j-1]:
                if result[lens-j-2] <= result[lens-j-1]:
                    result[lens-j-2] = max(result[lens-j-2], result[lens-j-1]) + 1
        for k in range(lens):
            candy += result[k]
        return candy