백준 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만으로 이루어져 있다.

출력

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

 

 

# 해결법

 

O가 연속될때마다 점수가 올라가는 것을 어떻게 구현해야 할지 헷갈렸다. 처음에는 아예 for문을 81번 돌리면서 'O'*i 를 해주고 그 인덱스를 점수로 활용하는게 어떨까 생각해봤는데 아무리 생각해도 더 나은 방법이 있을 것 같아 포기했다.

고민하다가 for문 내부에 점수를 담당하는 for문을 하나 더 생성해서 O를 만날때마다 1을 더하고 값을 sum에 저장하는 식으로 구현했다. X를 만날때는 cnt 값을 초기화시켜준다.

 

# 나의 코드

n = int(input())
for i in range (n):
    x = str(input())
    sum = 0
    cnt = 0
    for j in list(x):
        if j == 'O':
            cnt += 1
            sum += cnt
        else:
            cnt = 0
    print(sum)