백준 2506번 파이썬 문제풀이 : 점수계산

문제 링크 : https://www.acmicpc.net/problem/2506

2506번: 점수계산

문제

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	2	3	0	0	1	2	0

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

입력

첫째 줄에 문제의 개수 N (1 ≤ N ≤ 100)이 주어진다. 둘째 줄에는 N개 문제의 채점 결과를 나타내는 0 혹은 1이 빈 칸을 사이에 두고 주어진다. 0은 문제의 답이 틀린 경우이고, 1은 문제의 답이 맞는 경우이다. 

출력

첫째 줄에 입력에서 주어진 채점 결과에 대하여 가산점을 고려한 총 점수를 출력한다. 

 

 

# 문제 풀이 ( 브론즈 3 )

- 강의용 문제 풀이

n을 입력 받고 n에 해당하는 수 만큼 채점 결과를 입력 받아야 하는데 n은 입력만 받고 input().split을 통해 스페이스를 제거하고 채점 결과를 p에 입력한다.

 

N = int(input())
p = input().split()
score = 0
count = 0
for i in p:
    if i == "1":
        if count == 0:
            score += 1
            count = 1
 
        else:
            count += 1
            score += count
    elif i == "0":
        count = 0

만약 채점 결과가 1이고 count 가 0이면 바로 이전의 문제를 틀렸기 때문에 1점만 score에 추가를 한다. 하지만 count가 0이 아닐 경우 count를 먼저 1을 올려준 후 count 만큼 score를 올려주면 된다.

또한 채점 결과가 0이면 다음 문제는 score 를 1점만 올려야 하기 때문에 count 를 0으로 초기화 해준다.