알고리즘/백준

[백준] 8958번 OX퀴즈 (파이썬)

알감자 2021. 12. 26. 00:44
 

8958번: OX퀴즈

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

www.acmicpc.net

코드

import sys

n = int(input())
li = list(sys.stdin.readline().rstrip() for i in range(n))

for iox in li:
     sum = 0
     simpe_sum = 0
     for i in range(len(iox)):
          if i > 0:
               if iox[i] == "O":
                    if iox[i-1] == "O":
                         simpe_sum += 1
                         sum += simpe_sum
                    elif iox[i-1] == "X":
                         sum += 1
                         simpe_sum += 1
               elif iox[i] == "X":
                    simpe_sum = 0
          elif i <= 0:
               if iox[i] == "O":
                    sum += 1
                    simpe_sum += 1
               elif iox[i] == "X":
                    pass

     print(sum)