弱爆了的Candies

题目出处

题目描述：

n个小朋友坐成一排，每个小朋友有一个数表示他的表现(数字越大表现越好)。老师要给每个小朋友发至少1颗糖，相邻的两个小朋友，得分较高的小朋友必须得到更多的糖，问：老师至少需要给出多少颗糖

我的解法：

开两个数组c,c1。分别表示到这个数字的连续上升的序列长度，和连续下降的长度，那么发给第i个小朋友的糖果数量就是max(c[i],c1[i])

因此有挫逼代码如下：

def Solve():
    n = int(input())
    l = []
    for i in range(n):
        x = int(input())
        l.append(x)
    c = [1 for i in range(0,n)]
    for i in range(1,n):
        if ( l[i] > l[i-1] ):
            c[i] = c[i-1]+1
    c1 = [1 for i in range(0,n)]
    i = n-2
    while ( i >= 0 ):
        if ( l[i] > l[i+1] ):
            c1[i] = c1[i+1]+1
        i -= 1
    ans = 0
    for i in range(n):
        if ( c[i] > c1[i] ):
            ans += c[i]
        else:
            ans += c1[i]
    print(ans)

Solve()

坐等空间复杂的o(1)做法