分糖果问题进阶：NC CD78

题目描述：
一群孩子做游戏，现在请你根据游戏得分来发糖果，要求如下：

1. 每个孩子不管得分多少，起码分到一个糖果。
2. 任意两个相邻的孩子之间，得分较多的孩子必须拿多一些糖果。
3. 任意两个相邻的孩子之间的得分如果一样多，糖果数必须相同
   这个第三点是相对之前的问题增加的约束条件。
   给定一个数组arr代表得分数组，请返回最少需要多少糖果。

输入描述：
第一行一个整数N表示数组大小
接下来一行N个整数表示数组内的元素
输出描述：
输出一个整数表示答案

示例1
输入：
3
1 2 2
输出：
5
说明：最优分配方案为1, 2, 2
示例2
输入：
13
0 1 2 3 3 3 2 2 2 2 2 1 1
输出：
30
说明：最优的分配方案为 1 2 3 4 4 4 2 2 2 2 2 1 1

思路1：
和之前的问题一样，可以采取两边都走一遍规则的方式，然后取两个数组的最大值进行相加。
思路2：
同样地，利用递增递减的数量保存的方式，只不过是在相等的情况下，设置为相同的糖果数而不是重新设置为1；然后注意一下如果相等时正处于递减序列，需要将递减序列的数量+1。

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# pick candy
# @param arr int整型一维数组 the array
# @return int整型
#
n = int(input())
l = list(map(int, input().strip().split()))
pre, inc, dec = 1, 1, 0
res = 1
for i in range(1, n):
    if l[i] > l[i-1]:
        dec = 0
        pre += 1
        inc = pre
        res += pre
    elif l[i] == l[i-1]:
        if dec > 0:
            dec += 1
        res += pre
    else:
        dec += 1
        if dec == inc:
            dec += 1
        pre = 1
        res += dec
print(res)