Найдите, какое минимальное число фигурок ей понадобится, чтобы Мила смогла заполнить нижние ряды поля

[goldolov_na]

 

[regnor]

хорошо...

 

[goldolov_na]

хорошо...

n = int(input())
a = list(map(int, input().split()))
 
 
ans = 0
heights_stack = [0]
height_max = max(a)
a.append(height_max)
 
for i in range(n + 1):
    hole_height = height_max - a[i]
    while hole_height < heights_stack[-1]:
        ans += heights_stack.pop() - heights_stack[-1]
    if hole_height > heights_stack[-1]:
        heights_stack.append(hole_height)
print(ans)

написал код, работает только для примера в задании а для остальных примеров не работает

 

[regnor]

похоже на задачу с какой то олимпиадки...

 

[regnor]

совпадение?

Мила любит играть в Тетрис. Сегодня ей попалась новая игра, которая очень похожа на Тетрис.В - Школьные Знания.com

Школьные знания.com это сервис в котором пользователи бесплатно помогают друг другу с учебой, обмениваются знаниями, опытом и взглядами.

znanija.com