[프로그래머스] 다트 게임 - 파이썬

January 16, 2021

문제 설명

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

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

입출력 예제

내 코드

import re

def solution(dartResult):
    result = []
    scores = re.findall('\d+',dartResult)
    operands = re.findall('\D',dartResult)
    for operand in operands :
        if operand in ['S', 'D', 'T'] :
            power = 1 if operand == 'S' else 2 if operand == 'D' else 3
            num = ''.join(scores[:1])
            scores.remove(num)
            result.append(int(num) ** int(power))
            print(result)
        elif operand == '*' :
            if len(result) < 2 :
                result[0] *= 2
            elif len(result) == 2 :
                result[0] *= 2
                result[1] *= 2
            else :
                result[1] *= 2
                result[2] *= 2
        else :
            if len(result) == 1 :
                result[0] *= -1
            elif len(result) == 2 :
                result[1] *= -1
            else :
                result[2] *= -1
    return sum(result)

일단 처음에는 분기를 나누어 다 처리해보려고 했는데, 주어진 예시 테스트케이스에서 하나가 실패해서 보니 10이 문제였다. 점수가 0부터 10까지였기 때문에 주어진 문자열에서 하나씩 빼서 분기처리하게되면 10을 판별해내지 못했다. 그래서 어쩔 수 없이 문자열에서 숫자만 뺴는 걸 찾다보니 정규식인 re 모듈을 사용하였다. 정규식은 자유롭게 활용하기에는 잘 모르지만 숫자나 문자열을 뽑아내기 좋았다. \d+는 한글자가 아닌 붙은 숫자를 찾는거고, \D는 한글자짜리 문자를 말하는 것이다. 그렇게 점수와 Operand를 분리하여 추출한 후에 분기에서 활용했다. 일단 숫자면 제곱까지 마친 수를 result에 넣고 *이나 #이 나올 경우에는 result에 저장된 값을 활용하여 구하고 마지막에 result의 sum()을 구하여 답을 구했다.

다른 사람의 코드

import re


def solution(dartResult):
    bonus = {'S' : 1, 'D' : 2, 'T' : 3}
    option = {'' : 1, '*' : 2, '#' : -1}
    p = re.compile('(\d+)([SDT])([*#]?)')
    dart = p.findall(dartResult)
    for i in range(len(dart)):
        if dart[i][2] == '*' and i > 0:
            dart[i-1] *= 2
        dart[i] = int(dart[i][0]) ** bonus[dart[i][1]] * option[dart[i][2]]

    answer = sum(dart)
    return answer

가장 많은 좋아요를 받은 코드였는데 역시 10때문에 re를 쓴 것 같은데 나와는 다르게 findall()을 통해서 배열로 추출한 것이 아니고 compile()을 통해 정규식을 먼저 만들었다. 정규식을 통해 만든 예시는 [('1', 'D', '#'), ('2', 'S', '*'), ('3', 'S', '')]과 같은 형태로 만들어진다. 숫자가 먼저나오고 문자가 SDT 중에 나오고 *이랑 #은 ?를 활용해 있어도되고 없어도 되게끔 정규식을해서 만든 후에 미리 만들어놓은 딕셔너리에서 key로 value를 조회해서 계산식에 사용했다. 딕셔너리를 잘 쓰고, 정규식을 더 잘 활용해서 나보다 훨씬 간결한 코드가 나온 것 같다.