135.分发糖果

老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。

你需要按照以下要求，帮助老师给这些孩子分发糖果：

每个孩子至少分配到 1 个糖果。
相邻的孩子中，评分高的孩子必须获得更多的糖果。
那么这样下来，老师至少需要准备多少颗糖果呢？

链接：https://leetcode-cn.com/problems/candy

class Solution:
    def candy(self, ratings) -> int:
        n = len(ratings)
        sum = 1
        # inc: 递增序列的长度
        # dec: 递减序列的长度
        # pre: 上一个人分配的糖果数量
        inc, dec, pre = 1, 0, 1
        for i in range(1, n):
            if ratings[i] >= ratings[i-1]:
                dec = 0
                pre = (1 if ratings[i] == ratings[i-1] else pre + 1)
                sum += pre
                inc = pre
            else:
                dec += 1
                if dec == inc:
                    dec += 1
                sum += dec
                pre = 1
        return sum