hackerrank-candies.py

def candies(n, arr):
    if len(arr) == 1:
        return 1
    
    candies = [1]*len(arr)
    candies[0] = 1

    for i in range(1, len(arr)):
        if arr[i] > arr[i-1]:
            candies[i] = candies[i] + candies[i-1]
    for i in range(len(arr)-2, 0, -1):
        if arr[i] > arr[i-1] and candies[i] <= candies[i+1]:
            candies[i] = candies[i+1] + 1
    return sum(candies)