弱爆了的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)做法