[코딩 문제 풀이(파이썬/Python)] 프로그래머스 _ 다트게임

 

https://programmers.co.kr/learn/courses/30/lessons/17682

코딩테스트 연습 - [1차] 다트 게임

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

[문제 설명]

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

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

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

 

10을 놓쳐서 오래걸렸다. option을 제외한 숫자를 먼저 구해두고, option 연산을 진행했다.

def solution(dartResult):
    dic = ['S','D','T']
    answer = []
    dartResult = dartResult.replace('10','x')
    for i in range(len(dartResult)):
        if dartResult[i] in dic:
            if dartResult[i-1] == 'x':
                answer.append(pow(10, dic.index(dartResult[i])+1))
            else:
                answer.append(pow(int(dartResult[i-1]), dic.index(dartResult[i])+1))
    for i in range(len(dartResult)):
        num = i//2
        if num == 1:
            if dartResult[i] == '*':
                answer[0] *= 2 
            elif dartResult[i] == '#':
                answer[0] *=  - 1 
        elif num == 2:
            if dartResult[i] == '*':
                answer[0] *= 2
                answer[1] *= 2
            elif dartResult[i] == '#':
                answer[1] *=  - 1 
        elif (num == 3) or (num == 4):   
            if dartResult[i] == '*':
                answer[1] *= 2
                answer[2] *= 2
            elif dartResult[i] == '#':
                answer[2] *=  - 1 
    return sum(answer)

아래는 정규표현식을 이용한 풀이. 간결해보이기는 하나 이해하기 꽤 복잡하고, 연산 효율도 좋게 나타나지는 않았다. 그래도 가장 로직이 문제가 요구하는 것에 가까워보인다.

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

정규식에 사용된 메타문자 +와 ?는 각각 다음의 의미이다.

+: 앞서 표현된 정규식이 1개 이상 존재 (\d+ = 숫자 1개 이상)

?: 앞서 표현된 정규식이 있을 수도 있고, 없을 수도 있음 ([*#]? = * 혹은 #이 있거나 없음)