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

다트 게임

문제 설명

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인 경우가 엣지 케이스였는데 현재 문자가 0이고 바로 전 문자가 1이면 10으로 값을 업데이트 해줬다.

def solution(dartResult):
    answer = []
    score = {'S': 1, 'D': 2, 'T': 3}
    for i, c in enumerate(dartResult):
        if not c.isdigit():
            if c == '*':
                answer[-1] *= 2
                if len(answer) >= 2:
                    answer[-2] *= 2
            elif c == '#':
                answer[-1] = -answer[-1]
            else:
                answer[-1] = answer[-1] ** score[c]
        else:
            if c == '0' and dartResult[i-1] == '1':
                answer[-1] = 10
                continue
            answer.append(int(c))
    return sum(answer)

다른 사람 풀이

10을 먼저 처리하고 시작하는 방법도 있었다. replace로 10을 대체해놓고 각 문자열을 배열로 만들어 처리했다.

i를 -1로 시작해서 현재 값으로 업데이트해주면 이전 인덱스를 더 쉽게 확인해줄 수 있다.

def solution(dartResult):
    point = []
    answer = []
    dartResult = dartResult.replace('10','k')
    point = ['10' if i == 'k' else i for i in dartResult]
    print(point)
​
    i = -1
    sdt = ['S', 'D', 'T']
    for j in point:
        if j in sdt :
            answer[i] = answer[i] ** (sdt.index(j)+1)
        elif j == '*':
            answer[i] = answer[i] * 2
            if i != 0 :
                answer[i - 1] = answer[i - 1] * 2
        elif j == '#':
            answer[i] = answer[i] * (-1)
        else:
            answer.append(int(j))
            i += 1
    return sum(answer)

다른 사람 풀이 2

정규식을 이용한 풀이

import re
​
​
def solution(dartResult):
    bonus = {'S' : 1, 'D' : 2, 'T' : 3}
    option = {'' : 1, '*' : 2, '#' : -1}
    p = re.compile('(\d+)([SDT])([*#]?)') # 숫자 1개 이상 반복 SDT 중 하나와 일치 *#와 0또는 1회 일치
    dart = p.findall(dartResult)
    #[('1', 'S', ''), ('2', 'D', '*'), ('3', 'T', '')]
    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