[Python/알고리즘] 인프런 점수계산

문제출처

OX 문제는 맞거나 틀린 두 경우의 답을 가지는 문제를 말한다. 여러 개의 OX 문제로 만들어진 시험에서 연속적으로 답을 맞히는 경우에는 가산점을 주기 위해서 다음과 같이 점수 계산을 하기로 하였다. 1번 문제가 맞는 경우에는 1점으로 계산한다. 앞의 문제에 대해서는 답을 틀리다가 답이 맞는 처음 문제는 1점으로 계산한다. 또한, 연속으로 문제의 답이 맞는 경우에서 두 번째 문제는 2점, 세 번째 문제는 3점, ..., K번째 문제는 K점으로 계산한다. 틀린 문제는 0점으로 계산한다.

예를 들어, 아래와 같이 10 개의 OX 문제에서 답이 맞은 문제의 경우에는 1로 표시하고, 틀린 경우에는  0으로  표시하였을  때,  점수  계산은  아래  표와  같이  계산되어,  총  점수는 1+1+2+3+1+2=10 점이다.

풀이코드

import sys

n = int(input())

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

sum = 0
res = 0
for i in range(len(a)):
    if a[i] == 1:
        res += 1
        sum += res
    else:
        res = 0
print(sum)

코드설명

1. for문으로 리스트 원소에 맞게 0이면 res= 0, 1이면 1을 res에 저장
2. 연속적으로 맞다면 누적된 sum에 res를 저장