백준 8958번(python)

---

T = int(input())

listed_OX =[input() for _ in range(T)]

for OX in listed_OX:
    total_sum = 0
    X_split = OX.split('X')
    for correct in X_split:
        n = len(correct)
        
        if(n > 0):
            total_sum += (n * (n+1)) // 2
            
    print(total_sum)

 

+ 정규표현식 re module 사용

import re

T = int(input())

listed_OX = [input() for _ in range(T)]
p = re.compile('O+')

for OX in listed_OX:
    total_sum = 0
    X_split = p.findall(OX)
    
    for ele in X_split:
        n = len(ele)
        
        total_sum += (n * (n+1)) // 2
        
    print(total_sum)