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

Donghyun·2024년 7월 4일

코드카타 파이썬 프로그래머스

Code Kata - 파이썬

목록 보기

5/54

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

문제 설명

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

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

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

제한사항

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

입출력 예

s	result
"one4seveneight"	1478
"23four5six7"	234567
"2three45sixseven"	234567
"123"	123

입출력 예 설명

입출력 예 #1

문제 예시와 같습니다.

입출력 예 #2

문제 예시와 같습니다.

입출력 예 #3

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

입출력 예 #4

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

제한시간 안내

정확성 테스트 : 10초

문제풀이

일부 자릿수가 영단어로 바뀐 문자열을 보고 원래 숫자를 찾아 출력하면 되는데… 어떻게?

딕셔너리를 활용해볼까?

영단어를 key 로, 숫자를 값으로 가지는 딕셔너리 dict_1 을 만들고
dict_1.items() 로 key, value 에 접근
for 문을 돌면서
1. key 가 문자열(s) 에 있으면? → answer 에 value 추가
2. value 가 문자열(s) 에 있으면? → answer 에 value 추가
마지막으로 문자열 answer 를 정수형으로 바꿔주며 return

전체 구문 합친 코드

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

    answer = ''

    for i, value in dict_1.items():
        if i in s:
            answer += str(value)
        elif str(value) in s:
            answer += str(value)

    return int(answer)

코드 실행 → 4 개 중 4 개 성공!

오케이 ~ 쉽네 ~

근데, 제출 후 채점을 했더니… 정답률이 10% 가 나왔다. 뭐가 문제였을까? 내가 뭘 놓쳤을까?

내 코드의 문제점

내가 만든 코드는 1 ~ 9 의 순으로 문자열 s 를 탐색하며 숫자를 추가하는 방식이다. 입출력 예시로 주어진 문자열 s 의 예시들은 운이 좋게도 1 ~ 9 의 순서로 구성되어 있어서 코드를 실행 했을 때 문제가 없었던 것.

따라서, 일치하는 item, value 를 찾아 answer 에 추가하는 것이 아니라

s 에서 문자열에 해당하는 부분을 숫자로 대체해주는 작업이 필요.
이후 s 를 정수형으로 변환해서 return 하면 해결!

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

    for i, value in dict_1.items():
        s = s.replace(i, str(value))

    return int(s)