[Python] 백준 8958번 - OX퀴즈

OX퀴즈

문제 )

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

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

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

입력 )

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

출력 )

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

내 풀이 )

# 테스트 케이스 입력받기
T = int(input())

# 결과값 넣을 리스트 선언
result = []

# 테스트 케이스만큼 반복
for i in range(T):
    # OX 입력받아서 리스트에 저장
    ox_list = list(input())
    # 변수 초기화
    score = 0  
    sum_score = 0  
    # 입력받은 OX 길이만큼 반복
    for j in ox_list:
        if j == 'O':
            score += 1  
            sum_score += score  
        else:
            score = 0
    result.append(sum_score)
 
# 결과값 출력
for value in result:
    print(value)

 

해설 )

- score 변수는 O가 연속된 개수만큼 플러스, X를 만나면 score = 0으로 초기화

- sum_score는 총점

- 한 줄씩 계산한 후 result 리스트에 결과값 저장

- 한 줄에 하나씩 출력

결과 )

 

* 저는 테스트 케이스만큼 다 입력받고 결과를 한 번에 출력하도록 했는데

오른쪽 사진처럼 하나 입력하고 결과가 바로 나오게 해도 맞았다고 뜨긴 뜨더라구여,,ㅎㅎ;;

편한 방법대로 해보시면 될 것 같아요!