[프로그래머스] 2021 카카오 채용연계형 인턴십 01.숫자 문자열과 영단어

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

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

 

문제설명

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

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

• 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에는 영단어로 바뀐 부분이 없습니다.

 

풀이

문자열 관련된 문제로 영어 단어를 숫자로 변경하면 쉽게 풀 수 있는 문제이다.

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

dictionary 자료형을 이용하여 영어 단어를 숫자에 mapping하였다.
값을 정수형이 아니라 문자로 한 이유는 s에 이미 포함되어 있는 숫자도 문자로 들어오기 때문이다.

 

 

temp = ''
    for i in s:
        if '0' <= i <= '9':
            if temp != '':
                answer += numbers[temp]
                temp = ''
            answer += i
        else:
            temp += i
            if temp in numbers:
                answer += numbers[temp]
                temp = ''

문자열 s의 값을 받으면서 i값이 0 ~ 9면 temp에 해당하는 숫자를 answer에 추가해준다.
i값이 0 ~ 9가 아니라면 temp에 i값을 추가한 뒤 dictionary의 key 중에 있는지 검사하고 있다면
answer에 추가한다.

 

return int(answer)

마지막 answer에는 str형의 답이 쌓여있기 때문에 int로 형변환을 해주면 된다.

 

 

전체 코드

def solution(s):
    answer = ''
    numbers = {'zero': '0', 'one': '1', 'two': '2', 'three': '3', 'four': '4',
               'five': '5', 'six': '6', 'seven': '7', 'eight': '8', 'nine': '9'}

    temp = ''
    for i in s:
        if '0' <= i <= '9':
            if temp != '':
                answer += numbers[temp]
                temp = ''
            answer += i
        else:
            temp += i
            if temp in numbers:
                answer += numbers[temp]
                temp = ''
    return int(answer)