백준 8958 python

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

8958번: OX퀴즈

 

import sys

n = int(sys.stdin.readline().rstrip())  # 테스트 케이스의 개수

for i in range(n):
    score = 0  # 각 테스트 케이스의 점수
    total_score = 0  # 총 점수
    result = sys.stdin.readline().rstrip()  # OX 퀴즈의 결과
    
    for j in range(len(result)):
        if result[j] == 'O':
            score += 1
            total_score += score
        else:
            score = 0
    
    print(total_score)