[프로그래머스] 숫자 문자열과 영단어 python

문제

네오와 프로도가 숫자놀이를 하고 있습니다. 네오가 프로도에게 숫자를 건넬 때 일부 자릿수를 영단어로 바꾼 카드를 건네주면 프로도는 원래 숫자를 찾는 게임입니다.

다음은 숫자의 일부 자릿수를 영단어로 바꾸는 예시입니다.

• 1478 → "one4seveneight"
• 234567 → "23four5six7"
• 10203 → "1zerotwozero3"

이렇게 숫자의 일부 자릿수가 영단어로 바뀌어졌거나, 혹은 바뀌지 않고 그대로인 문자열 s가 매개변수로 주어집니다. s가 의미하는 원래 숫자를 return 하도록 solution 함수를 완성해주세요.

풀이

wordDictionary = {
    "zero" : 0,
    "one" : 1,
    "two" : 2,
    "three" : 3,
    "four" : 4,
    "five" : 5,
    "six" : 6,
    "seven" : 7,
    "eight" : 8,
    "nine" : 9
}
def solution(s):
    if s.isnumeric():
        return int(s)
    word = ''
    answer = ''
    for char in s:
        if char.isnumeric():
            answer += char
        else:
            word += char
            if word in wordDictionary.keys():
                answer += str(wordDictionary[word])
                word = ''
    return int(answer)

영단어를 key로, 대응되는 숫자를 value로 가진 사전을 하나 만들어두었습니다.

주어진 문자열 s가 모두 숫자면 바로 리턴을 하고, 영어라면 문자를 하나씩 더해 딕셔너리의 키가 되는지 확인하고, 된다면 해당 문자에 해당하는 숫자를 answer에 추가해주었습니다. 이렇게 했던 이유는 oneseven과 같이 영어가 붙어있을 경우, 확인하기 위해서였습니다.

 

간결한 풀이

num_dic = {"zero":"0", "one":"1", "two":"2", "three":"3", "four":"4", "five":"5", "six":"6", "seven":"7", "eight":"8", "nine":"9"}

def solution(s):
    answer = s
    for key, value in num_dic.items():
        answer = answer.replace(key, value)
    return int(answer)

 

코딩테스트 연습 - 숫자 문자열과 영단어