b = 0
c = 0
while True:
    a = int(input())
    if b + a == 0 and a != 0:
        c += a ** 2
        break
    else:
        b += a
        c += a ** 2
print©
			
		Chelik00101Если по шагам всё пройдёшь вместе с кодом, то получишь два.
хотя должен быть 0, разве нет?
>>> def f(): ... b = 0 ... c = 0 ... ... while True: ... a = int(input()) ... if b + a == 0 and a != 0: ... c += a ** 2 ... break ... else: ... b += a ... c += a ** 2 ... ... print(c) ... >>> f() -1 1 2 >>>