https://www.hackerrank.com/challenges/candies/problem

candies

#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the candies function below.
def candies(n, arr):
    n = len(arr)
    dp = [1]*n
    sum = 0

    for i in range(1, n, 1):
        if arr[i] > arr[i-1]:
            dp[i] = dp[i] + dp[i-1]

    for i in range(n-2, -1, -1):
        if arr[i] > arr[i+1] and dp[i] <= dp[i+1]:
            dp[i] = dp[i+1] + 1

    for d in dp:
        sum += d

    return sum

if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    n = int(input())

    arr = []

    for _ in range(n):
        arr_item = int(input())
        arr.append(arr_item)

    result = candies(n, arr)

    fptr.write(str(result) + '\n')

    fptr.close()