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

 

문제 설명

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

카카오톡 게임별의 하반기 신규 서비스로 다트 게임을 출시하기로 했다. 다트 게임은 다트판에 다트를 세 차례 던져 그 점수의 합계로 실력을 겨루는 게임으로, 모두가 간단히 즐길 수 있다.

갓 입사한 무지는 코딩 실력을 인정받아 게임의 핵심 부분인 점수 계산 로직을 맡게 되었다. 다트 게임의 점수 계산 로직은 아래와 같다.

1.다트 게임은 총 3번의 기회로 구성된다.

2.각 기회마다 얻을 수 있는 점수는 0점에서 10점까지이다.

3.점수와 함께 Single(S), Double(D), Triple(T) 영역이 존재하고 각 영역 당첨 시 점수에서 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	1^1 * 2 + 2^2 * 2 + 3^2
2	1D2S#10S	9	1^2 + 2^1 * (-1) + 10^1
3	1D2S0T	3	1^2 + 2^1 + 0^3
4	1S*2T*3S	23	1^1 * 2 * 2 + 2^3 * 2 + 3^1
5	1D#2S*3S	5	1^2 * (-1) * 2 + 2^1 * 2 + 3^1
6	1T2D3D#	-4	1^3 + 2^2 + 3^2 * (-1)
7	1D2S3T*	59	1^2 + 2^1 * 2 + 3^3 * 2

접근법

오래 고민했는데 생각보다 어렵지 않은 문제였습니다. 먼저 저는 반복문을 통해 문자열의 문자를 하나씩 접근하면서 정수인 경우에는 value_list에, 옵션인 경우에는 option_list에, 보너스인 경우에는 bonus_list에 넣었습니다.

-먼저 숫자를 찾을 때는 숫자가 두 자리의 수 일수도 있기 때문에 현재 판별하고있는 문자의 다음 문자가 숫자일 경우 두 개의 문자를 합친 뒤, 정수형으로 변환하여 리스트에 넣어주었습니다. 이 때, 두 번째 자리 수 정수 차례가 오면 다시 리스트에 저장될 수 있기 때문에 현재 판별하고 있는 문자의 이전 문자가 숫자일 경우는 continue시켜 주었습니다.

-옵션인 경우에는 리스트의 값을 1로 세팅해 주고, #인 경우에는 -1을 곱해주고 *인 경우에는 value에 들어있는 수의 자릿값을 기준으로 직전 점수까지 2를 곱해줍니다. (점수가 무조건 먼저 나오고 option이 뒤에 나오기 때문에 value_list에 들어있는 수의 자릿값을 기준으로합니다. 예를들어서 value_list에 [1,2] 이 있을 경우 option_list의 값으로는 [2,2]가 들어오게 됩니다.) 또한 처음 점수를 받고 나서 보너스 *를 얻을 경우는 따로 처리해 줍니다.

-마지막으로 보너스인 경우에도 value_list의 인덱스를 이용하여 S는 1, D는 2, T는 3을 bonus_list에 저장해 줍니다.

 

나의 코드

bonus = ['S','D','T']
option = ['#','*']
def solution(dartResult):
    answer = 0
    value_list = []
    option_list = [1,1,1]
    bonus_list = []
    for i,char in enumerate(dartResult):
        if not char.isdigit():
            if char in bonus:
                bonus_list.append(bonus.index(char)+1)
            else:
                if char is '#': option_list[len(value_list)-1] *= -1
                else:
                    if len(value_list) == 1: option_list[0] *= 2
                    else:
                        option_list[len(value_list)-1] *= 2
                        option_list[len(value_list)-2] *= 2
        if char.isdigit():
            if i > 0 and dartResult[i-1].isdigit(): continue
            if dartResult[i+1].isdigit(): value_list.append(int(char+dartResult[i+1]))
            else: value_list.append(int(char))
    for i,value in enumerate(value_list):
        answer += pow(value,bonus_list[i])*option_list[i]
    return answer

다른 코드

import re
def solution(dartResult):
    shot = re.findall(r'\d{1,2}[SDT][*#]?', dartResult)
    opt = [1,1,1]
    for i, s in enumerate(shot):
        if s[-1] == '#':
            opt[i] *= -1
            shot[i] = shot[i][:-1]
        elif s[-1] == '*':
            opt[i] *= 2
            shot[i] = shot[i][:-1]
            if i:
                opt[i-1] *= 2
    point = [(int(s[:-1]) ** '0SDT'.find(s[-1]) * o) for s, o in zip(shot, opt)]
    return sum(point)

위의 방법은 re모듈을 사용하는 방법인데, 점수|보너스|옵션으로 이루어져있기 때문에, 정규 표현식을 이용하여 나눠줄 수 있습니다. \d{1,2}는 한개 또는 두개인 정수를 의미하고, [SDT]는 하나의 알파벳(보너스)에 대응하고, [*#]또한 하나의 문자(옵션)에 대응합니다. 하지만 있을수도 있고, 없을수도 있기 때문에 ?를 붙여줍니다.