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

백준 파이썬 > 8958번 OX퀴즈

출처 : 백준 8958번 OX퀴즈

8958번: OX퀴즈

---

나의 풀이 in Python3

n = int(input())
for i in range(n):
    ox = input()
    l = list(ox)
    sum = 0
    c =1
    for k in l:
        if k == 'O':
            sum += c
            c +=1
        else:
            c = 1
    print(sum)

OX 테스트 케이스를 리스트화 시키고, 반복문으로 "O"가 나올 때마다 sum 값에 1부터 등차수열로 값을 더해준다.

 

그렇게 어렵지는 않았으나 깔끔하게 풀지 못해 다른 사람의 풀이를 공유해봅니다.

 

다른 사람의 풀이 in Python3

from sys import stdin
for _ in range(int(stdin.readline())):
    print(sum(sum(range(1,len(i)+1)) for i in stdin.readline().strip().split(sep='X')))