코딩테스트[파이썬]/알고리즘 문제풀이 입문

[구현력 키우기] - 점수계산

softmoca__ 2024. 2. 5. 12:58
목차

 

점수계산

 

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

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

 

 

 

 

나의 코드

n=int(input())

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

res=0
cnt=0
for i in range(n):
    if arr[i]==1:
        cnt+=1
        res+=cnt
    else:
        cnt=0

print(res)

점수의 가중치를 계산하는 cnt변수를 사용해서 1이면 가중치를 늘리고 0이면 가중치를 초기화시켰다.

 

 

 

정답 코드

import sys
sys.stdin=open("input.txt", "r")
n=int(input())
a=list(map(int, input().split()))
cnt=0
sum=0
for i in range(n):
    if a[i]==1:
        cnt=cnt+1
        sum=sum+cnt
    else:
        cnt=0

print(sum)

나의 코드와 일치 한다 !