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

문제 보기

https://school.programmers.co.kr/learn/courses/30/lessons/81301

프로그래머스

소스 코드

def solution(s):
    answer = ''
    dic = {'zero':'0', 'one':'1', 'two':'2', 'three':'3', 'four':'4',
          'five':'5', 'six':'6', 'seven':'7', 'eight':'8', 'nine':'9'}
    
    tmp = ''
    for i in s:
        tmp += i
        if tmp in dic.keys():
            answer += dic[tmp]
            tmp = ''
        elif tmp in dic.values():
            answer += tmp
            tmp = ''
            
    return int(answer)

 

 

개선 코드

def solution(s):
    answer = ''
    words = ['zero', 'one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine']
    
    for i in range(len(words)):
        s = s.replace(words[i], str(i))
    
    return int(s)