백준 2506 점수계산

n = int(input())
ox = list(map(int, input().split()))  #문제에 대한 OX

k = 0  #정답의 연속성 체크
sum = 0  #합계
for i in range(n):
    if ox[i] == 1: 
        k += 1
        sum += k
    elif ox[i] == 0: 
        k = 0

print(sum)

2023-03-09