Итак, пишу олимпиаду, в питоне недавно. Компилятор пишет ошибка в строчке 6. Вроде все верно, можете помочь и сказать какая ошибка?
a = 0
n = int(input())
p = 3
while n>0:
c = int(input())
if c == 1:
a = a+p
p = p+1
if c == 0:
p = max(3, p-3)
n-=1
print(a)
a = 0
n = int(input())
p = 3
while n>0:
c = int(input())
if c == 1:
a = a+p
p = p+1
if c == 0:
p = max(3, p-3)
n-=1
print(a)