programmers | Lv1. 다트 게임 [Python]

💡 Python 3

🔗 문제

[1차] 다트 게임 [Link]

💻 코드

def solution(dR):
    dR = dR.replace('10', 'N')
    result = []
    for i, s in enumerate(dR):
        if s == 'N': s = '10'     
        if s.isdigit():
            k = int(s)
            if dR[i+1] == 'D': k **= 2
            elif dR[i+1] == 'T': k **= 3
            result.append(k)
            
        if i ==2 and s == '*':
            r1 = result.pop()*2
            result.append(r1)
        elif i != 2 and s == '*':
            r1 = result.pop()*2
            r2 = result.pop()*2
            result.append(r2)
            result.append(r1)
        elif s == '#':
            result.append(result.pop()*(-1))
    return sum(result)

💥 다른 사람 코드

정규표현식 활용.

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

• \d+: 하나 이상 연결된 숫자

  • \d: 숫자 1글자
  • + : 하나 이상

• findall(): 정규식과 매치되는 모든 문자열을 리스트형식으로 리턴

글만봐서는 무슨 소린지 잘 모르겠어서 print 찍어봤다.

dR = "1S2D*3T"
def solution(dartResult):
    import re
    bonus = {'S' : 1, 'D' : 2, 'T' : 3}
    option = {'' : 1, '*' : 2, '#' : -1}
    p = re.compile('(\d+)([SDT])([*#]?)')
    dart = p.findall(dartResult)
	print(dart)

# output
	[('1', 'S', ''), ('2', 'D', '*'), ('3', 'T', '')]

참고 자료

[파이썬] 정규표현식(regular expression)
정규표현식 #2
04 모든 문자열 패턴 찾기 (findall)