[문자열] 프로그래머스 lv1. 다트게임 (정규표현식, 10치환, isdecimal, 숫자찾기)

문제

문제 바로가기

다트 게임은 총 3번의 기회로 구성된다. 각 기회마다 얻을 수 있는 점수는 0점에서 10점까지이다. 점수와 함께 Single(S), Double(D), Triple(T) 영역이 존재하고 각 영역 당첨 시 점수에서 1제곱, 2제곱, 3제곱 (점수1 , 점수2 , 점수3 )으로 계산된다. 옵션으로 스타상(*) , 아차상(#)이 존재하며 스타상(*) 당첨 시 해당 점수와 바로 전에 얻은 점수를 각 2배로 만든다. 아차상(#) 당첨 시 해당 점수는 마이너스된다. 스타상(*)은 첫 번째 기회에서도 나올 수 있다. 이 경우 첫 번째 스타상(*)의 점수만 2배가 된다. (예제 4번 참고) 스타상(*)의 효과는 다른 스타상(*)의 효과와 중첩될 수 있다. 이 경우 중첩된 스타상(*) 점수는 4배가 된다. (예제 4번 참고) 스타상(*)의 효과는 아차상(#)의 효과와 중첩될 수 있다. 이 경우 중첩된 아차상(#)의 점수는 -2배가 된다. (예제 5번 참고) Single(S), Double(D), Triple(T)은 점수마다 하나씩 존재한다. 스타상(*), 아차상(#)은 점수마다 둘 중 하나만 존재할 수 있으며, 존재하지 않을 수도 있다. 0~10의 정수와 문자 S, D, T, *, #로 구성된 문자열이 입력될 시 총점수를 반환하는 함수를 작성하라.

입력 형식
"점수|보너스|[옵션]"으로 이루어진 문자열 3세트.
예) 1S2D*3T

점수는 0에서 10 사이의 정수이다.
보너스는 S, D, T 중 하나이다.
옵선은 *이나 # 중 하나이며, 없을 수도 있다.

코드 전개

1. 문자열을 숫자와 bonus, option 각각으로 쪼개서 새로운 배열로 만들어준다. (isdecimal 사용)
   주의, 숫자가 두자리 수 일 수 있다.
2. 문자열을 돌면서, 덧셈할 항을 total 리스트에 추가해준다. (sum(total)이 답이 되도록)
3. bonus는 이전항이 무조건 숫자이므로, 이전항에 제곱(**1, **2, **3)을 해서 total에 추가해준다.
4. option은 이전항 혹은 전전항이 있다면 두개를 꺼내서 각각 *2 혹은 *1을 해서 다시 total에 추가해준다.
5. sum(total)

코드

def solution(dart):

    bonus = ['S','D','T']
    opt = ['*','#']

    lst = []
    n = ''
    for i in range(len(dart)):
        d = dart[i]
        if d.isdecimal():
            print(d)
            n += d
        else:
            if n:
                lst.append(n)
                n=''
            lst.append(d)

    total = []

    for idx in range(1,len(lst)):
        d = lst[idx]
        if d in bonus:
            n = int(lst[idx-1]) ** (bonus.index(d)+1)
            total.append(n)
        elif d in opt:
            if d == '*':
                if len(total)>=2:
                    n = total.pop()*2
                    m = total.pop()*2
                    total.append(m)
                    total.append(n)
                else:
                    n = total.pop()*2
                    total.append(n)
            if d == '#':
                n = total.pop()*(-1)
                total.append(n)


    return (sum(total))

그 외

정규표현식을 쓰고 싶었는데, 잘 안됐다. 다른 분 코드보니, 역시나 정규표현식 쓴 게 있다...
별도 포스팅이 필요하다. 정규표현식.
정규표현식 없이도 충분히 짤 수 있었는데, 그 이유가 두 자리 숫자가 10밖에 없기 때문이다.

참고해야 할 코드

정규표현식 사용 (re.compile, findall)

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을 별도 문자로 치환해놓고, list로 바꿀 때 다시 10으로 복원해줌.. 천재)

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)

배열의 마지막 두 놈을 어떻게 바꿀지 고민이었는데, 이분은 list[-2:]부분을 대체했다. 길이 검증 안해도 되는구나

def solution(dartResult):
    dart = {'S':1, 'D':2, 'T':3}
    scores = []
    n = 0

    for i, d in enumerate(dartResult):
        if d in dart:
            scores.append(int(dartResult[n:i])**dart[d])
        if d == "*":
            scores[-2:] = [x*2 for x in scores[-2:]]
        if d == "#":
            scores[-1] = (-1)*scores[-1]
        if not (d.isnumeric()):
            n = i+1

    return sum(scores)