candies

gistfile1.rb

N = gets.to_i
a = [0]*(N)


for i in 0..(N-1) do
   a[i] = gets.to_i
end

n = [1]*(N)

for i in 0..(N-2) do
   if a[i] > a[i+1] and n[i] <= n[i+1]
        n[i] = n[i+1] + 1 
   elsif a[i] < a[i+1] and n[i] >= n[i+1]
        n[i+1] = n[i] + 1 
   end
end

(N-1).downto(1) do |i|
   if a[i] > a[i-1] and n[i] <= n[i-1]
        n[i] = n[i-1] + 1 
   elsif a[i] < a[i-1] and n[i] >= n[i-1]
        n[i-1] = n[i] + 1
   end
end

t = 0

for i in 0..(N-1) do
  t = t + n[i].to_i
end

print t