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

2021년 5월 17일 Algorithm, 프로그래머스, 다트 게임

2018년 카카오 블라인드 채용 1차 문제

TL;DR

• 주어진 문자열을 조건에 따라 토큰화(문자열 처리)할 수 있는지
• 문제에 주어진 조건을 구현할 수 있는지
• 정규식을 활용할 수 있는지

* 문제 분석

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

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

 

- 2 ~ 4 : 숫자(0~10)-보너스(S,D,T)-옵션(*,#)의 형태로 토큰을 나누어야 한다.

- 5 ~ 8 : 구현해야 하는 보너스와 옵션의 동작을 명시하고 있다. 문제에서 주어진 대로 코드에 반영하면 된다.

- 9 : 정규식을 적용할 경우, 중요한 힌트가 된다.

 

* 입-출력 형태

문제의 입출력 형태

 

- dartResult : 입력으로 주어지는 문자열이다.

- answer : 문자열을 통해 얻을 수 있는, 출력해야 하는 점수이다.

- 설명 : 문제 조건을 예시를 통해 설명하고 있다. 4번 예시에서는 6번 조건에 대해, 5번 예시에서는 7번의 조건을 보여주고 있다.

 

* 풀이

NUMBERS = '0123456789'
BONUS = 'SDT'
OPTIONS = '*#'

def solution(dartResult):
    dartResult = list(dartResult)
    score = 0
    scores = []

    while len(dartResult) != 0:
        item = dartResult.pop(0) # 인식해야 하는 토큰의 종류를 확인한다.
        if item in NUMBERS: # 토큰이 점수(숫자)인 경우를 처리하는 구문이다.
            score = int(item)
            if score == 1: # 10을 처리하기 위한 구문이다.
                if dartResult[0].isdigit():
                    score = 10
                    dartResult.pop(0) # 10은 두 개의 문자가 합쳐지므로 0까지 제거해준다.

        if item in BONUS: # 토큰이 보너스인 경우를 처리하는 구문이다.
            if item == 'S':
                scores.append(score)
            elif item == 'D':
                scores.append(score ** 2)
            else:
                scores.append(score ** 3)

        if item in OPTIONS: # 토큰이 옵션인 경우를 처리하는 구문이다.
            if item == '#':
                scores[-1] = -1 * scores[-1]
            else:
                if len(scores) == 1: # 스타가 처음에 나오는 경우를 처리하기 위한 구문이다.
                    scores[-1] = 2 * scores[-1]
                else:
                    scores[-2] = 2 * scores[-2]
                    scores[-1] = 2 * scores[-1]

    return sum(scores)

 

- 주어진 문자열의 각 원소를 조회하며, 문제에서 주어진 조건을 그대로 구현하였다.

- 점수가 10인 경우, 2개의 문자가 하나의 토큰을 이루기 때문에 이를 처리하는 부분을 구현하였다.

- 스타상의 경우 가장 처음부터 나올 수 있는데 이를 위한 조건을 구현하였다.

 

* 개선된 풀이

- 정규식을 활용할 경우, 훨씬 더 간단하게 코드를 구현할 수 있다.

- 파이썬에서는 re 모듈을 통해 정규 표현식을 지원한다.

 

re 모듈 출력의 예시

 

- line 2 : 제시된 정규식과 같이 주어진 문자열을 나누어주는 역할을 한다.

• \\d+ : 점수를 인식한다. +가 붙은 이유는 10이 2개의 숫자가 필요하기 때문이다.
• \[SDT\] : 보너스를 인식한다. S, D, T중 하나일 경우가 이에 해당한다.
• \[\*#\]? : 옵션을 인식한다. *, # 중 하나일 경우가 이에 해당한다. ?가 붙은 이유는 9번 조건, 옵션은 없을 수도 있다는 것 때문이다.

- line 3 ~ 5 : 입력에 정규식을 적용하면 나오는 출력을 확인할 수 있다. 각 정규식에 해당하는 요소로 튜플을 구성하여, 이를 리스트에 저장한다.

- 이를 적용하여 문제를 다음과 같이 해결할 수 있다.

 

import re

def solution(dartResult):
    BONUS = {'S': 1, 'D': 2, 'T': 3}
    OPTION = {'': 1, '*': 2, '#': -1}

    scores = []

    p = re.compile('(\d+)([SDT])([*#]?)')
    dart = p.findall(dartResult)
    for token in dart:
        if token[2] == '*' and len(scores) > 0:
            scores[-1] *= 2
        scores.append(int(token[0]) ** BONUS[token[1]] * OPTION[token[2]])

    return sum(scores)