[Python/Bronze2/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만으로 이루어져 있다 .

 

|    출력

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

 

 

---

정답 코드

import sys
input = sys.stdin.readline
from collections import deque

class Solution:
    def oxquiz(self):
        n = int(input())
        for _ in range(n):
            Q = deque(input().rstrip())
            i, res = 0,0
            while Q:
                x = Q.popleft()
                if x =='O':
                    i += 1
                    res += i
                else:
                    i = 0
            print(res)

if __name__ == "__main__":
    s = Solution()
    s.oxquiz()

 

 

|    비고

처음 생각난 건 deque인데 굳이 큐말고 스택으로 풀어도 될 것 같다.