[프로그래머스 lv1] 숫자 문자열과 영단어(파이썬)

2021 카카오 채용연계형 인턴십 문제입니다.

 

문제 설명

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

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

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

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

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

숫자에 대응되는 영단어

제한사항

• 1 ≤ s의 길이 ≤ 50
• s가 "zero" 또는 "0"으로 시작하는 경우는 주어지지 않습니다.
• return 값이 1 이상 2,000,000,000 이하의 정수가 되는 올바른 입력만 s로 주어집니다.

입출력 예

입출력 예

입출력 예 설명

입출력 예 #1

• 문제 예시와 같습니다.

입출력 예 #2

• 문제 예시와 같습니다.

입출력 예 #3

• "three"는 3, "six"는 6, "seven"은 7에 대응되기 때문에 정답은 입출력 예 #2와 같은 234567이 됩니다.
• 입출력 예 #2와 #3과 같이 같은 정답을 가리키는 문자열이 여러 가지가 나올 수 있습니다.

입출력 예 #4

• s에는 영단어로 바뀐 부분이 없습니다.

제한시간 안내

• 정확성 테스트 : 10초

---

 

 

 

나의 풀이

import re

def solution(s):
    values = [str(n) for n in range(0, 10)]
    keys = ['zero', 'one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine']
    eng = dict(zip(keys, values))
    
    for i in eng.keys():
        s = re.sub(f'{i}', f'{eng[i]}', s)
    return int(s)

딱 보자마자 아주 간단한 치환 문제로 생각할 수 있다.

각 숫자에 대응되는 단어가 있기 때문에 딕셔너리를 사용했다.

eng = { } 안에 직접 키와 값을 입력해도 되고, 위와 같이 키와 값을 각각 리스트로 만들어서 zip으로 묶어 한번에 만들어 줄 수도 있다.

 

그리고 사전을 순회하며 영문자로 쓰여진 것을 숫자로 바꿔주면 된다.

 

이 과정을 문자열 내장함수인 replace를 사용해도 되고, 나의 풀이처럼 정규식을 사용해도 된다.

 

리턴할 때는 문자열인 s를 정수로 바꿔서 리턴하면 된다.

 

테스트 결과