백준 8958번 <OX퀴즈> [파이썬]

https://www.acmicpc.net/problem/8958

8958번: OX퀴즈

문제

"OOXXOXXOOO"와 같은 OX퀴즈의 결과가 있다. O는 문제를 맞은 것이고, X는 문제를 틀린 것이다. 문제를 맞은 경우 그 문제의 점수는 그 문제까지 연속된 O의 개수가 된다. 예를 들어, 10번 문제의 점수는 3이 된다.

"OOXXOXXOOO"의 점수는 1+2+0+0+1+0+0+1+2+3 = 10점이다.

OX퀴즈의 결과가 주어졌을 때, 점수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 길이가 0보다 크고 80보다 작은 문자열이 주어진다. 문자열은 O와 X만으로 이루어져 있다.

출력

각 테스트 케이스마다 점수를 출력한다.

예제 입력 1 

5
OOXXOXXOOO
OOXXOOXXOO
OXOXOXOXOXOXOX
OOOOOOOOOO
OOOOXOOOOXOOOOX

예제 출력 1 

10
9
7
55
30

 

 

가장 헷갈린 건 일단 점수를 주는 방식이 가장 헷갈렸다. 누구나 그랬겠지만.

그래서 생각한건 O가 나올 때 무조건 1점을 받고 그 뒤에 O나 X가 없으면 어차피 점수를 주거나 할 수 없기 때문에 예외처리를 조금 해두는

식으로 진행해 봤다.

import sys

n = int(sys.stdin.readline())
list_a = []
list_b = []
for i in range(n):
    a = sys.stdin.readline()
    list_a.append(a)

global point
global cnt
for i in range(n):
    point = 0
    cnt = 1
    length = len(list_a[i])
    for j in range(length):
        if list_a[i][j] == 'O':
            point = point + 1
            if j + 1 != length and list_a[i][j+1] == 'O':
                point = point + cnt
                cnt = cnt + 1
            else:
                cnt = 1

    list_b.append(point)

for i in range(len(list_b)):
    print(list_b[i])

비어있는 list 2개를 만들어 두고 한 리스트에는 ox를 받고 다른 리스트에 그 ox 점수를 받는 형식으로 진행했다.

보시다시피

 

for i in range(n):
    point = 0
    cnt = 1
    length = len(list_a[i])
    for j in range(length):
        if list_a[i][j] == 'O':
            point = point + 1
            if j + 1 != length and list_a[i][j+1] == 'O':
                point = point + cnt
                cnt = cnt + 1
            else:
                cnt = 1
    list_b.append(point)

이 부분이 점수를 주는 for문 형식의 알고리즘인데

변수 cnt를 통해 연속적으로 O가 나올 것을 해결했다.

 

아직 헷갈리는게 많지만, 계속 이렇게 글을 써가면서 반복 학습을 하는 것 같아 좋다.

한번에 말고 꾸준히!!