코딩테스트 연습/백준

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

슈퍼짱짱 2019. 8. 5. 13:43
반응형

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

 

8958번: OX퀴즈

문제 "OOXXOXXOOO"와 같은 OX퀴즈의 결과가 있다. O는 문제를 맞은 것이고, X는 문제를 틀린 것이다. 문제를 맞은 경우 그 문제의 점수는 그 문제까지 연속된 O의 개수가 된다. 예를 들어, 10번 문제의 점수는 3이 된다. "OOXXOXXOOO"의 점수는 1+2+0+0+1+0+0+1+2+3 = 10점이다. OX퀴즈의 결과가 주어졌을 때, 점수를 구하는 프로그램을 작성하시오. 입력 첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는

www.acmicpc.net


코드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

반응형