카테고리 없음

[프로그래머스 / Python] [1차]다트 게임(2018 KAKAO) - Lv.1

wookhyung 2021. 8. 5. 00:07
728x90

[1차]다트 게임(2018 KAKAO) - Lv.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

입출력 예제

예제 dartResult answer 설명
1 1S2D*3T 37 11 * 2 + 22 * 2 + 33
2 1D2S#10S 9 12 + 21 * (-1) + 101
3 1D2S0T 3 12 + 21 + 03
4 1S*2T*3S 23 11 * 2 * 2 + 23 * 2 + 31
5 1D#2S*3S 5 12 * (-1) * 2 + 21 * 2 + 31
6 1T2D3D# -4 13 + 22 + 32 * (-1)
7 1D2S3T* 59 12 + 21 * 2 + 33 * 2

풀이

def solution(dartResult):
    dic = {"S": 1, "D": 2, "T": 3}
    dartResult = dartResult.replace("10", "X")
    stack = []

    for i in dartResult:
        if i.isdigit() or i == "X":
            stack.append(10 if i == "X" else int(i))
        elif i in ["S", "D", "T"]:
            num = stack.pop()
            stack.append(num ** dic[i])
        elif i == "#":
            stack[-1] *= -1
        elif i == "*":
            num = stack.pop()
            if len(stack):
                stack[-1] *= 2
            stack.append(2 * num)


    return sum(stack)

3가지 결과 값을 하나하나 계산하면서 stack에 집어넣고 결과값을 모두 더하는 식으로 해결했다.

 

먼저, 10점이 존재하면 for문에서 1과 0이 따로 나오기 때문에 다른 문자(X)로 바꾼 뒤 이후에 if문에서 X가 나오면 10으로 바꾸었다.

 

또한 보너스 점수에 S, D, T는 각각 1제곱, 2제곱, 3제곱인데 해당 문자가 나오면 숫자로 바꿔주기 위해서 dict를 사용했다.

 

이후에는 for문을 돌게 되는데,

 

만약 i가 숫자이면 스택에 넣고 X라면 10으로 바꿔서 스택에 넣는다.

 

i가 보너스 dict에 해당하는 문자라면 스택 안에 있는 맨 마지막 값을 pop으로 꺼내서 dict[i]의 값만큼 제곱해준다.

 

i가 #이라면 마지막 값에 -1를 곱하고,

*이라면 마지막 값을 꺼내고 그래도 stack에 요소가 남아있으면 그 값에 2를 곱해주고, 꺼냈던 마지막 값에 또 2를 곱해준 뒤 다시 추가해준다.

 

해당 for문이 끝나면 스택에는 3가지 숫자가 남아있을 것이고 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