[백준] 17389 보너스 점수 with Python

📌 BOJ 17389 보너스 점수

💡 조건

1. OX표에 N개의 문제들이 있을 때, 1번 문제, 2번 문제, ..., N번 문제 순으로 채점된다.
   문제는 뒤로 갈수록 어려워지기 때문에, i 번 문제의 기본 점수는 i 점이다.

2. 문제를 맞히면 그 문제의 기본 점수(즉 i 번 문제의 경우 i 점)를 획득하며, 틀리면 얻지 못한다.
   기본 점수와 별개로, '보너스 점수'라는 값이 존재한다. 이는 처음에는 0점이다.

3. 문제를 맞히면 그 때의 '보너스 점수'를 획득하고, '보너스 점수'의 값이 1점 증가한다.
   문제를 틀리면 '보너스 점수'를 얻지 못하고, '보너스 점수'의 값이 0점으로 초기화된다.

4. 첫 번째 줄에 OX표의 길이인 자연수 N이 주어진다. (1 ≤ N ≤ 10,000)

5. 두 번째 줄에 OX표를 의미하는 문자열 S가 주어진다.
   S는 O(알파벳 대문자 O, ASCII 코드 79)와 X(알파벳 대문자 X, ASCII 코드 88)로만 구성되어 있으며, 길이는 N이다.

6. 문자열 S의 i 번째 글자가 O이면 해당 참가자가 i 번째 문제를 맞혔음을 의미하고, X이면 틀렸음을 의미한다.

7. 구현 유형의 문제

🔖 예제 및 실행결과

예제 1

8
XOOOXOOX

실행결과 1

26

⌨️ 문제 풀이

1. 문제의 조건을 잘 살펴본 뒤, 예제를 살펴보자.
   문제를 맞히면 그 문제의 기본 점수(즉 i 번 문제의 경우 i 점)를 획득하며, 틀리면 얻지 못한다.
   기본 점수와 별개로, '보너스 점수'라는 값이 존재한다. 이는 처음에는 0점이다.

   문제를 맞히면 그 때의 '보너스 점수'를 획득하고, '보너스 점수'의 값이 1점 증가한다.
   문제를 틀리면 '보너스 점수'를 얻지 못하고, '보너스 점수'의 값이 0점으로 초기화된다.

2. 예제에 대한 설명을 보며 구현을 하면 된다.

🖥 소스 코드

from sys import stdin

n = int(stdin.readline())
s = stdin.readline().rstrip()
ans = 0
cnt, flag = 0, 0

for i in range(len(s)):
    if s[i] == 'X':
        cnt = 0
    else:
        if flag:
            cnt += 1
            ans += (i + 1) + cnt
            continue

        ans += (i + 1) + cnt
        cnt += 1
print(ans)