[백준] 8958 : OX퀴즈 (Python)

string형태로 입력받아 s에 저장하고 s에 값을 하나하나씩 확인하며 O일 경우 count를 1증가하고 X일 경우에는 count를 0으로 초기화한다.
result는 모든 count를 합한 값을 가지게 된다.

나의코드

n = int(input())
result = 0

for i in range(n):
    s = str(input())
    result = 0
    count = 0
    for j in range(len(s)):
        if s[j] == 'O':
            count += 1
        else:
            count = 0
        result += count
    print(result)