Candies

1def candies(n, arr):
2  # Write your code here
3  # candies[0] = 1
4  # loop through arr using i from (1, len(arr)):
5  #   if arr[i - 1] < arr[i]: candies[i] = candies[i] + 1
6  #   else: candies[i] = 1
7  cand = [1 for _ in range(len(arr))]
8  for i in range(1, len(arr)):
9    if arr[i - 1] < arr[i]: cand[i] = cand[i - 1] + 1
10  
11  for i in range(len(arr) - 2, -1, -1):
12    if arr[i] > arr[i + 1] and cand[i] <= cand[i + 1]: cand[i] += (cand[i + 1] - cand[i]) + 1
13  
14  return sum(cand)