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

출력

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

제출 코드

n = int(input())
for i in range(n):
  cnt = 0
  sum = 0
  ox = input()
  for j in ox:
    if j == "O":
      cnt += 1
      sum += cnt
    elif j == "X":
      cnt = 0
  print(sum)

1. 반복할 횟수를 정수로 입력 받는다.
2. 반복 횟수만큼 OX 배열을 입력 받아 ox에 문자열로 저장한다.
3. 문자열을 입력 받을 때마다 cnt, sum 변수를 생성하고 초기화 한다. cnt는 각 "O"의 점수를 뜻하고 sum은 점수의 합을 담는다.
4. 문자열을 처음부터 비교해 "O"가 나오면 cnt를 1씩 증가시키다가 "X"가 나오면 값을 다시 0으로 초기화하도록 만든다.
5. 그리고 각 문자열마다 cnt의 값을 더하여 sum에 담고 한 문자열에 대해서 작업이 끝나면 sum을 출력한다.