[백준-python] 8958번 : OX퀴즈

https://www.acmicpc.net/problem/8958

8958번: OX퀴즈

---

코드1

n=int(input()) # 테스트 케이스 개수
for cnt in range(n): # 테스트 케이스 만큼 반복
    a=input() # 테스트 케이스 입력
    arr=[0]*len(a) # 각 문제의 점수를 담을 array
    if(a[0]=="O"):arr[0]=1 # 1번문제 점수
    for i in range(1,len(a)):
        if(a[i]=="O"): # 해당 문제가 맞았을 경우 
            arr[i] = arr[i-1]+1 # 전 점수 +1
    print(sum(arr)) # 출력

>>> a=input()

>? OOXXOXXOOO

>>> arr=[0]*len(a)
       if(a[0]=="O"):arr[0]=1
       for i in range(1,len(a)):
            if(a[i]=="O"):
                arr[i] = arr[i-1]+1

>>> arr

[1, 2, 0, 0, 1, 0, 0, 1, 2, 3]

>>> sum(arr)

10

---

코드2

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')))

>>> print(sum(sum(range(1,len(i)+1)) for i in stdin.readline().strip().split(sep='X')))

>? OOXXOXXOOO

10

 

>>> print([i for i in stdin.readline().strip().split(sep='X')])

>? OOXXOXXOOO

['OO', '', 'O', '', 'OOO']

 

>>> print([sum(range(1,len(i)+1)) for i in stdin.readline().strip().split(sep='X')])

>? OOXXOXXOOO

[3, 0, 1, 0, 6]

 

---

sys.stdin : https://leedakyeong.tistory.com/8