Помогите решить задачу на цикле while. python

c =  0

n = 0

while True:

   n = int(input())

   if n == 0:

       break

   c += 1

print(c)

Ответ:

Незнайка записался в отряд космонавтов и готовился к полёту на Луну. Вращаясь на центрифуге он слушает называемые числа и считает их, пока не назовут число ноль. Числа после нуля он считать не должен. Помогите Незнайке пройти испытание.

Решение:

n, i = 1,0

while n != 0:

   n = int(input())

   if n != 0:

       i += 1

print(i)