题解 | #分糖果问题#

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# pick candy
# @param arr int整型一维数组 the array
# @return int整型
#
class Solution:
    def candy(self , arr: List[int]) -> int:
        # write code here
        left=[1 for i in range(len(arr))]
        right=[1 for i in range(len(arr))]
        n=len(arr)
        ans=0
        for i in range(1,n):
            if arr[i]>arr[i-1]:
                left[i]=left[i-1]+1
        for j in range(n-2,-1,-1):
            if arr[j]>arr[j+1]:
                right[j]=right[j+1]+1
        for i in range(n):
            ans+=max(left[i],right[i])
        return ans