카카오 신입 공채 1차 코딩 테스트 문제 2. 다트 게임(난이도: 하) (Python)

2017/11/03 - [공부] - 카카오 신입 공채 1차 코딩 테스트 문제 2. 다트 게임(난이도: 하)

Java에 이어 Python으로도 작성해 봤다.

import re
 
 
def question2(in_str):
    digits = re.findall('\d+', in_str)
    non_digits = re.findall('\D+', in_str)
    score = []
 
    for i in range(0, 3):
        score.append(int(digits[i]))
        if non_digits[i].count('S'):
            None
        elif non_digits[i].count('D'):
            score[i] = score[i] * score[i]
        elif non_digits[i].count('T'):
            score[i] = score[i] * score[i] * score[i]
 
        if non_digits[i].count('#'):
            score[i] = score[i] * -1
        elif non_digits[i].count('*'):
            if i < 1:
                score[i] = score[i] * 2
            else:
                score[i - 1] = score[i - 1] * 2
                score[i] = score[i] * 2
 
    total_score = 0
    for s in score:
        total_score += s
    return total_score