section2.10 점수 계산

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 0 1 1 1 0 0 1 1 0 점수 1 0 1 2 3 0 0 1 2 0 시험문제의 채점 결과가 주어졌을 때, 총점수를 계산하는 프로그램을 작성하시오.

 

 

# 점수 계산
cnt 이용해서 풀이하면 될 듯

import sys
sys.stdin = open("input.txt", "rt")
n = int(input())
result = list(map(int, input().split()))
cnt = 0
price = 0

for i in range(n):
    if (cnt==0 and result[i]==1):
        cnt += 1
    elif (cnt==0 and result[i]==0):
        cnt = 0
    elif (cnt!=0 and result[i]==1):
        cnt +=1
    else:
        cnt = 0
    price += cnt

print(price)

 

이 문제도 앞의 문제와 같이 경우의 수만 나누면 되는 문제였다

 

import sys
#sys.stdin = open("input.txt", "rt")
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)

 

아쉬운 점은 당연히 현재의 인덱스보다 앞에 있는 숫자가 1인지 0인지를 확인하고 cnt를 초기화해야한다고 생각했는데 아니었다

생각보다 더 쉬운 문제였고 내가 만든 코드는 굳이였다