[Python] 파이썬 8958번, 파이썬 문자열 저장 방식

OX퀴즈 (정답률 51%)

문제

"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

Source Code

def plus(num):  # 수열의 합 구하는 함수
    r = 0
    for i in range(1, num + 1):
        r += i
    return r


result = 0

for _ in range(int(input())):
    o = 0
    quiz = input()
    for q in quiz:
        if q == 'O':  # 연속으로 맞은 개수 구하기
            o += 1
        else:  # X가 나오면 연속으로 맞은 개수 점수 계산
            result += plus(o)
            o = 0
    result += plus(o)  # 마지막 문제가 정답이면 계산이 되지 않아서 마지막 처리
    print(result)
    result = 0

O의 갯수를 세다가 X가 나오면 연속적인 정답이 끊긴다.

그 때 점수 계산을 하는데 마지막문제가 O이면 점수 계산을 하지 못하고 for문이 끝나서 끝난 후에 다시 계산을 하도록 만들었다.

파이썬의 문자열은 끝에 NULL문자나 공백문자가 저장되지 않고 for문을 숫자의 형태로 잘 사용하지 않기 때문에 생각할 때 주의를 해야될 것 같다. ex) a = "abc" 이면 코드 그대로 "abc"만 저장됨.