[프로그래머스 코딩테스트] lv1. 숫자 문자열과 영단어

이 문제는 2021 카카오 채용연계형 인턴십 채용을 위한 코테 문제 중 하나입니다. 생각보다 간결해 금방 풀 수 있을 것 같습니다!

 

문제설명

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

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

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

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

참고로 각 숫자에 대응되는 영단어는 다음 표와 같습니다.

숫자영단어

0	zero
1	one
2	two
3	three
4	four
5	five
6	six
7	seven
8	eight
9	nine

 

나의 풀이

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

저는 s를 하나씩 읽어가며 answer에 nums에 해당하는 문자열 value들을 추가하는 형식으로 진행했습니다. 다른 사람들의 풀이를 보면 nums를 for loop 조건문에 넣어 s의 문자열들을 replace함수로 바꿔 훨씬 더 간결하게 코드를 짜놨습니다. 이처럼 저의 코드를 개선한 것이 다음 코드 블락과 같습니다.

def solution(s):
    
    nums = {
        'zero' : 0,
        'one' : 1,
        'two' : 2,
        'three' : 3,
        'four' : 4,
        'five' : 5,
        'six' : 6,
        'seven' : 7,
        'eight' : 8,
        'nine': 9
    }
    
    for num in nums:
        s = s.replace(num, str(nums.get(num)))

    return int(s)

 

문제 출처: https://programmers.co.kr/learn/courses/30/lessons/81301