Python 알고리즘 - OX퀴즈

OX퀴즈

• O가 나오면 1씩 증가하고, X가 나오면 0으로 초기화되어 그 수들의 합을 구하는 문제다. 나는 초기화가 가능한 count와 총합을 구하는 result_count를 선언해줬다.

N = int(input())
num_list = []
for i in range(N):
    count = 0 #O가 중첩될 때 늘어나는 값
    result_count = 0 # 최종결과값
    OX_num = list(input())
    for i in range(len(OX_num)):
        if OX_num[i] == 'O':
            count += 1
            result_count += count
        elif OX_num[i] == 'X':
            count = 0
    num_list.append(result_count)

for i in range(N):
    print(num_list[i])