백준 알고리즘 OX퀴즈 8958번

문제

"OOXXOXXOOO"와 같은 OX퀴즈의 결과가 있다. O는 문제를 맞은 것이고, X는 문제를 틀린 것이다. 문제를 맞은 경우 그 문제의 점수는 그 문제까지 연속된 O의 개수가 된다. 예를 들어, 10번 문제의 점수는 3이 된다.

"OOXXOXXOOO"의 점수는 1+2+0+0+1+0+0+1+2+3 = 10점이다.

OX퀴즈의 결과가 주어졌을 때, 점수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 길이가 0보다 크고 80보다 작은 문자열이 주어진다. 문자열은 O와 X만으로 이루어져 있다.

출력

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

예제 입력 

5
OOXXOXXOOO
OOXXOOXXOO
OXOXOXOXOXOXOX
OOOOOOOOOO
OOOOXOOOOXOOOOX

예제 출력 

10
9
7
55
30

OX 중 X는 값을 계산하는데 필요 없으므로 입력 받을 때 X를 기준으로 O를 받는다.

리스트에는 [ 'OO', '', 'OOOO', '', 'O' ] 이런식으로 원소가 들어있다.

OO라면 1+2, OOOO이라면 1+2+3+4

소스코드

C = int(input())

while C>0: #테스트 케이스

    sum = 0 #합을 0으로

    OX = list(map(str, input().split('X'))) #X로 입력된 값을 구분하여 리스트에 넣음

    for i in range(len(OX)): #리스트의 길이만큼 반복

        if OX[i] != '': #리스트에 있는 값이 공백이 아니라면

            for j in range(1, len(OX[i])+1): #1~리스트의 길이만큼 반복

                sum += j #j를 누적해서 더함

    C -= 1

    print(sum)