[프로그래머스] 다트 게임

[문제 설명]

다트 게임 카카오톡에 뜬 네 번째 별! 심심할 땐? 카카오톡 게임별~

카카오톡 게임별의 하반기 신규 서비스로 다트 게임을 출시하기로 했다. 다트 게임은 다트판에 다트를 세 차례 던져 그 점수의 합계로 실력을 겨루는 게임으로, 모두가 간단히 즐길 수 있다. 갓 입사한 무지는 코딩 실력을 인정받아 게임의 핵심 부분인 점수 계산 로직을 맡게 되었다. 다트 게임의 점수 계산 로직은 아래와 같다. 

1. 다트 게임은 총 3번의 기회로 구성된다.
2. 각 기회마다 얻을 수 있는 점수는 0점에서 10점까지이다.
3. 점수와 함께 Single(S), Double(D), Triple(T) 영역이 존재하고 각 영역 당첨 시 점수에서 1제곱, 2제곱, 3제곱 (점수1 , 점수2 , 점수3 )으로 계산된다.
4. 옵션으로 스타상(*) , 아차상(#)이 존재하며 스타상(*) 당첨 시 해당 점수와 바로 전에 얻은 점수를 각 2배로 만든다. 아차상(#) 당첨 시 해당 점수는 마이너스된다.
5. 스타상(*)은 첫 번째 기회에서도 나올 수 있다. 이 경우 첫 번째 스타상(*)의 점수만 2배가 된다. (예제 4번 참고)
6. 스타상(*)의 효과는 다른 스타상(*)의 효과와 중첩될 수 있다. 이 경우 중첩된 스타상(*) 점수는 4배가 된다. (예제 4번 참고)
7. 스타상(*)의 효과는 아차상(#)의 효과와 중첩될 수 있다. 이 경우 중첩된 아차상(#)의 점수는 -2배가 된다. (예제 5번 참고)
8. Single(S), Double(D), Triple(T)은 점수마다 하나씩 존재한다.
9. 스타상(*), 아차상(#)은 점수마다 둘 중 하나만 존재할 수 있으며, 존재하지 않을 수도 있다.

0~10의 정수와 문자 S, D, T, *, #로 구성된 문자열이 입력될 시 총점수를 반환하는 함수를 작성하라.

 

입력 형식

"점수|보너스|[옵션]"으로 이루어진 문자열 3세트.

예) 1S2D*3T

• 점수는 0에서 10 사이의 정수이다.
• 보너스는 S, D, T 중 하나이다.
• 옵선은 *이나 # 중 하나이며, 없을 수도 있다.

출력 형식 

3번의 기회에서 얻은 점수 합계에 해당하는 정수값을 출력한다. 

예) 37

 

[코드]

def solution(dartResult):
    scores = []
    i = 0

    while i < len(dartResult):
        if dartResult[i:i+2] == "10":
            score = 10
            i += 2
        else:
            score = int(dartResult[i])
            i += 1

        bonus = dartResult[i]

        if bonus == "S":
            score = score ** 1
        elif bonus == "D":
            score = score ** 2
        elif bonus == "T":
            score = score ** 3

        i += 1

        if i < len(dartResult) and dartResult[i] in "*#":
            option = dartResult[i]

            if option == "*":
                score *= 2

                if scores:
                    scores[-1] *= 2

            elif option == "#":
                score *= -1

            i += 1

        scores.append(score)

    return sum(scores)

이 문제는 문자열로 주어진 다트 결과를 해석해서 총점을 구하는 문제이다.

다트 결과는 점수 + 보너스 + 옵션 형태로 총 3번 반복된다.

여기서 핵심은 다음과 같다.

1. 점수를 읽는다.

2. 보너스 S,D,T에 따라 제곱한다.

3. 옵션 *, #이 있으면 적용한다.

4. 계산된 점수를 배열에 저장한다.

5. 마지막에 배열의 합을 return 한다.

 

먼저 각 기회에서 계산된 점수를 저장할 배열 scores를 만든다.

그리고 문자열을 앞에서부터 하나씩 확인하기 위해 인덱스 i를 사용한다.

 

다트 점수는 0부터 10까지 가능한데, 주의할 점은 10이다.

예를 들어 문자열이 10S라면 1과 0을 따로 보면 안 된다.

그렇기 때문에 따로 처리해준다.

if dartResult[i:i+2] == "10":
    score = 10
    i += 2
else:
    score = int(dartResult[i])
    i += 1

 

점수를 읽은 뒤에는 바로 뒤에 있는 보너스를 확인한다.

bonus = dartResult[i]

 

보너스에 따라 점수를 제곱한다.

if bonus == "S":
    score = score ** 1
elif bonus == "D":
    score = score ** 2
elif bonus == "T":
    score = score ** 3

 

보너스 처리가 끝난 후 인덱스를 한 칸 이동한다.

i += 1

 

그 다음에는 옵션이 있는지 확인한다.

옵션은 있을 수도 없을 수도 있기 때문에, 먼저 인덱스가 문자열 범위를 넘지 않는지 확인해야 한다.

if i < len(dartResult) and dartResult[i] in "*#":

 

옵션이 *라면 현재 점수를 2배로 만들고, 이전 점수도 존재한다면 이전 점수까지 2배로 만든다.

if option == "*":
    score *= 2

    if scores:
        scores[-1] *= 2

 

옵션이 #이라면 현재 점수를 음수로 만든다.

elif option == "#":
    score *= -1

 

옵션 처리가 끝났으면 인덱스를 한 칸 이동한다.

i += 1

 

현재 기회의 점수 계산이 끝났으므로 scores 배열에 저장한다.

scores.append(score)

 

모든 문자열을 다 확인하면 점수들의 합을 return 한다.

return sum(scores)

 

예를 들어 1S2D*3T의 경우 계산 과정은 다음과 같다.

1S  -> 1
2D* -> 4점이지만 * 때문에 현재 점수 8, 이전 점수 2
3T  -> 27

 

따라서 scores는 다음과 같다.

[2, 8, 27]

 

총합은 다음과 같다.

37

 

[개선사항]

def solution(dartResult):
    scores = []
    num = ""

    bonus_power = {
        "S": 1,
        "D": 2,
        "T": 3
    }

    for ch in dartResult:
        if ch.isdigit():
            num += ch

        elif ch in "SDT":
            scores.append(int(num) ** bonus_power[ch])
            num = ""

        elif ch == "*":
            scores[-1] *= 2

            if len(scores) >= 2:
                scores[-2] *= 2

        elif ch == "#":
            scores[-1] *= -1

    return sum(scores)

이 방식은 문자열을 한 글자씩 보면서 처리한다.

숫자가 나오면 num에 모아두고, S, D, T가 나오면 그때 점수를 계산해서 scores에 넣는다.

특히 10도 1, 0을 차례대로 num에 붙이기 때문에 따로 예외처리를 많이 하지 않아도 된다.

 

[시간복잡도]

dartResult 문자열의 길이를 N이라고 하면, 문자열을 한 번만 순회하므로 시간복잡도는 O(N)이다.