[BAEKJOON]백준 8958번: 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만으로 이루어져 있다.

 

출력

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

 

코드 

num = int(input())

for _ in range(num):
   quiz = list(map(str, input()))
   result = 0
   score = 0

   for i in range(len(quiz)):
       if quiz[i] == 'O':
           score += 1
           result += score
       elif quiz[i] == 'X':
           score = 0

   print(result)

코드 실행 결과

코드 설명

테스트 개수를 num에 입력받고 그 범위만큼 for문을 반복해준다.

quiz에 OX문자열을 입력받고, result와 score를 0으로 정의해준다.

num = int(input())

for _ in range(num):
   quiz = list(map(str, input()))
   result = 0
   score = 0

 

for문을 통해 quiz의 각 요소들을 체크한다.

O이면 score에 1씩 더한다. 

O이 연속될 경우 score이 1점씩 커질 것이다.

최종 결과인 result에 score점수를 더해준다.

만약 X가 나오면 score을 0으로 초기화한다.

result를 출력하고 프로그램을 종료한다.

   for i in range(len(quiz)):
       if quiz[i] == 'O':
           score += 1
           result += score
       elif quiz[i] == 'X':
           score = 0

   print(result)