[Python] 8958번 OX퀴즈 - 백준

OX퀴즈

8958번: OX퀴즈

1차원 배열

 

문제

"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

5
OOXXOXXOOO
OOXXOOXXOO
OXOXOXOXOXOXOX
OOOOOOOOOO
OOOOXOOOOXOOOOX

예제 출력 1

10
9
7
55
30

---

 

풀이

로그인

for _ in range(int(input())):
    l = input().split(sep='X')
    print(sum(sum(range(1, len(s) + 1)) for s in l))

 

X에는 점수가 없고, 연속된 O에 가산점이 붙는다. 각 줄을 X를 기준으로 분리시키면 연속된 O의 문자열로 이루어진 list가 생성된다. 이어지는 O의 길이만큼 range로 점수를 생성하여 합하는 것을 분리시킨 만큼 반복하면 쉽게 합계를 구할 수 있다.

 

추가적으로 아래의 방법을 사용해도 동일한 값을 얻을 수 있다.

sum(range(len(s))) + len(s)