分糖果问题进阶：NC CD78_lwfwcgluc3rhbgwtcgitmjq2ode1odg1ntat-优快云博客

本文介绍了一种算法解决孩子们游戏得分糖果分配问题，要求相邻得分高的孩子拿更多糖果且相等时数量一致。通过两种思路分析，演示如何计算最少所需糖果数量，适用于不同得分数组。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

每个孩子不管得分多少，起码分到一个糖果。
任意两个相邻的孩子之间，得分较多的孩子必须拿多一些糖果。
任意两个相邻的孩子之间的得分如果一样多，糖果数必须相同
这个第三点是相对之前的问题增加的约束条件。
给定一个数组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)