[백준 2506, Python] 점수 계산

# 백준 2506 점수계산

import sys
inpit = sys.stdin.readline

n = int(input()) # 문제의 개수 입력
sum = 0 # 점수를 받아 줄 변수 생성 
res = 0 # sum을 받아 줄 변수 생성 //sum이 0이 되어 오류가 날 경우를 위해 (14행)
a = list(map(int,input().split())) # 채점 결과 입력

for i in range(n): # n번 반복
  if a[i] == 1: # 만약 1이 나오면
    sum += 1 # sum에 +1
    res += sum # res에 sum을 더해줌
  else: # 아니면
    sum = 0 # sum = 0
print(res) # res출력