[BOJ] 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

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

acmicpc.net/problem/2506

입력

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

출력

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

예제

---

풀이

1번 풀이

1. OX결과와 길이가 같은 점수 리스트를 만든다
2. OX 결과가 X이면 점수는 0점이고 이전 결과값 prev도 False가 된다
3. OX 결과가 O이고 이전 결과가 False이면 점수는 1점이고 prev를 True로 바꾼다
4. OX 결과가 O이고 이전 결과가 True이면 점수는 이전 점수+1 이다

2번 풀이

1. 더해줄 점수score와 최종 출력값 ans를 선언한다
2. OX 결과가 O이면 score에 1을 더하고 score를 ans에 더한다
3. OX 결과가 X이면 score를 0으로 초기화하고 score에 ans를 더한다

---

코드

1번

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

score = [0] * N
prev = 0
for i in range(N):
    if not data[i]:
        score[i] = 0
        prev = 0
    elif not prev:
        score[i] = 1
        prev = 1
    else:
        score[i] = score[i-1] + 1
print(sum(score))

2번

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

score = ans = 0
for i in range(N):
    if data[i]:
        score += 1
    else:
        score = 0
    ans += score
print(ans)

---

후기

비교	1번 결과	2번 결과
시간	76ms	68ms
코드 길이	276B	181B
메모리	30864KB	30860KB