[python] 백준 8958번_OX퀴즈

출처: www.acmicpc.net/problem/8958

8958번: OX퀴즈

OX퀴즈 성공출처다국어분류  

시간 제한	메모리 제한	제출	정답	맞은 사람	정답 비율
1 초	128 MB	70529	35636	30581	51.747%

문제

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

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

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

입력

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

출력

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

예제 입력 1

5

OOXXOXXOOO

OOXXOOXXOO

OXOXOXOXOXOXOX

OOOOOOOOOO

OOOOXOOOOXOOOOX

예제 출력 1

10

9

7

55

30

---

연속된 O의 개수를 어떻게 구할지 가장 고민했던 것 같다. 

먼저 입력 첫번째 줄에서 테스트케이스 개수를 지정해주는 변수 N을 선언했다. 그리고 N으로 for 반복문을 통해

해당하는 OX값을 지정해주었고 문제 점수와 count변수를 0으로 초기화 시켜주었다.

 

그 다음 OX를 지정한 result 입력값에 대해 if 문을 넣어 해당 값이 'O'일 경우, count변수를 1 증가해주고 그 값을

sum 변수에 더해준다. 'X'일 경우에는 연속된 'O'가 끝난 상태이므로 count값을 0으로 다시 바꿔주었고, 이러한 과정을 반복하여 결과적으로 sum값을 출력하였다.

 

N=int(input())
for i in range(N):
    result=input()
    sum=0
    count=0
    for i in result:
        if i=='O':
            count+=1
            sum+=count
        else:
            count=0
    print(sum)

고민되는 부분을 해결하니 문제가 쉽게 풀렸다.