[알각코] 인프런 2-10 점수계산

점수계산

문제 정보는 인프런의 파이썬 알고리즘 문제풀이 코딩테스트에 있습니다!

문제풀이

n = int(input())
a = list(map(int, input().split()))

sum = 0
cnt = 0
for x in a:
  if x == 1:
    cnt += 1
    sum += cnt
  else:
    cnt = 0

print(sum)

풀때는 변수를 두 가지나 생각하지 못해서, sum, cnt 두 가지 예시를 보고 바로 풀엇당 ㅎㅎ…