Здравствуйте!
Есть такая задача:
Светофор, установленный на повороте в торговый центр «МЕГА» с Новомосковского тракта, работает в таком режиме, что за одну минуту повернуть успевают k машин. По выходным, когда жители города едут закупаться продуктами и одеждой, перед этим поворотом вырастает огромная пробка. Администрация торгового центра дала указание повесить под строящимся неподалёку мостом камеру, подсчитывающую количество автомобилей, подъезжающих к этому повороту со стороны города. По информации, полученной с камеры за n минут, прошедших с начала наблюдений, определите количество машин, стоящих в пробке в настоящий момент времени.
Исходные данные:
В первой строке записаны целые числа k и n — количество машин, успевающих повернуть на «МЕГУ» в течение минуты, и количество минут, прошедших с начала наблюдений. Во второй строке через пробел записаны целые числа
, …,
, где
— количество машин, подъехавших к повороту со стороны города в течение i-й минуты. Можно считать, что наблюдения начинаются рано утром, когда машин, ожидающих на повороте, ещё нет.
Результат:
Выведите количество машин, стоящих в пробке на повороте через n минут после начала наблюдений.
Вышло такое решение:
import sys
a=[]
for x in sys.stdin:
for y in x.split():
a.append(int(y))
n=0
for i in range(2, a[1]+2):
if n<=a[0]:
n=0
else:
n-=a[0]
if a[i]>a[0]:
n+=a[i]
if n<=a[0]:
print(0)
else:
print(n-a[0])
Но это решение не работает.
Где у меня ошибка?
