reegttg

reegttg

def candy_distribution(arr):
n = len(arr)
candies = [1] * n
for i in range(1, n):
if arr[i] > arr[i-1]:
candies[i] = candies[i-1] + 1
for i in range(n-2, -1, -1):
if arr[i] > arr[i+1]:
candies[i] = max(candies[i], candies[i+1] + 1)
return sum(candies)

arr = [1, 4, 3, 2]
print(candy_distribution(arr))

Scroll to Top