[백준/파이썬] 32642번 당구 좀 치자 제발

-Code

n = int(input())
weather = list(map(int, input().split()))
now, res = 0, 0

for el in weather:
    if el == 1:
        now += 1
    else:
        now -= 1

    res += now

print(res)