[알고리즘]점수계산

문제

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

1 0 1 1 1 0 0 1 1 0

시험문제의 채점 결과가 주어졌을 때, 총 점수를 계산하는 프로그램을 작성하시오.

입력예제1

10
1 0 1 1 1 0 0 1 1 0

출력예제1

10

[내 코드]

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

def sum(x):
    sum=0
    for num in x:
        sum+=num
    return sum

res=[]
count=0
i=0
while i<len(a):
    if a[i]==1:
        res.append(1)
        i+=1
        count=1
        while i<len(a) and a[i]!=0:
            count+=1
            i+=1
            res.append(count)
        count=0
    elif a[i]==0:
        res.append(0)
        i+=1
        
print(sum(res))

1. 각 리스트 요소들의 합을 구하는 함수 sum정의
2. 1번째 while문을 이용해서 a리스트의 원소가 1이 나오면 res리스트에 우선 1삽입 count=1, 2번째 while문을 통해 1다음 수가 1일 경우 count+1증가 후 res리스트에 삽입.원소가 0일 경우 2번째 while문 종료
3. 원소가 0일 경우 res리스트에 0삽입

[해설코드]

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)