백준_2506번

2506 점수계산

OX 문제 중 맞으면 1점 이후 맞을 때마다 점수는 가산된다.
하지만 틀리면 0점 이후 정답은 다시 1점 부터 계산한다는 문제.

파이썬에서는 첫번째 입력하는 문제수는 중요하지 않은 것 같다.

n = int(input())
x = list(map(int,input().split()))
n = 0
m = []

for i in x:
    if i == 1: 
        n += 1 # o 일 경우 점수를 쌓고
        m.append(n) # 맞은 문제의 점수를 리스트에 추가
    else: # x 일 경우 다시 0점으로 리셋
        n = 0

print(sum(m)) # 리스트의 점수를 sum 함수로 합쳐서 출력

문제를 간단하게 가져오고 싶었지만 결국 캡쳐본으로 ...ㅎㅎㅎ