백준 알고리즘 - 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만으로 이루어져 있다.

---

출력

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

 

---

접근법

1. 테스트 케이스 개수 입력
2. 테스트 케이스 한줄씩 입력
3. 한줄씩 OX 검사하기
   1. X 라면 더하지 않기
   2. O 라면 증가하면서 더하기

O일 때 누적해서 더해야 하고

X일 때 더하고 있는 값을 다시 0으로 초기화해야 한다.

 

어느 부분에서 초기화 해야 할지, 어느 부분에서 누적해서 더한 값을 print해주어야 할지 아는 것이 중요한 문제이다.

---

나의 코드

n = int(input())
OX = [list(map(str, input())) for _ in range (n)] # 한줄씩 문자열 입력받기

for i in range(len(OX)):
    add = 0 # 누적해서 더해주어야할 값
    answer = 0 # 점수
    for j in range(len(OX[i])):
        if OX[i][j] == 'X':
            add = 0 # X가 나왔을 때 초기화
        else:
            add += 1 # O가 나왔을 때 누적해서 1더하기
        answer += add
    print(answer)