[Python] Lv1. 숫자 문자열과 영단어

https://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로 주어집니다.

 

나의 접근 방법 및 풀이

문제 자체만 보면 영단어를 대응되는 숫자로 바꾸는 것이지만, 숫자와 영단어가 섞여있는 문자열 내에서 바꾸는 것이라 처음에는 어렵게 느껴졌다. 

하지만 대응 이라는 단어를 보고 딕셔너리를 생각하게 됐다.

표에 나와 있는 숫자와 이에 대응하는 영단어를 딕셔너리로 만들고 for()을 이용하여 딕셔너리 num_str의 key가 s에 존재한다면 replace()를 이용해서 key값을 대응되는 value, 즉 숫자로 바꾸었다.

그리고 s는 문자열이기때문에 마지막에 int() 형으로 형 변환을 해주어 return 하였다.

 

# Key:영단어, Value=숫자
num_str = {'zero':'0', 'one':'1','two':'2', 'three':'3','four':'4', 'five':'5','six':'6', 'seven':'7','eight':'8', 'nine':'9'}
def solution(s):
    for i in num_str:
        if i in s:
            s = s.replace(i, num_str[i])
    return int(s)

 

 

다른 풀이

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

def solution(s):
    answer = s
    for key, value in num_dic.items():
        answer = answer.replace(key, value)
    return int(answer)

if문을 사용하지 않고 바로 replace()를 사용하였다. 딕셔너리와 replace()를 사용했지만 코드가 더 간결해 보였다. 그런데 이경우는 시간 복잡도가 커질 수 있다는 댓글을 보았다.

 

def solution(s):
    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)

딕셔너리를 사용하지 않고 리스트를 이용하였다.

replace()를 이용하였지만, key와 value를 이용했던 딕셔너리와 다르게 words라는 리스트의 원소값과 해당 원소의 인덱스를 이용하여 문제를 풀었다.

 

 

 

 

 

 

카카오 문제라고 못 풀거라고 처음부터 좌절하지 말고 하나씩 해보는 노력을 해야겠다. 같은 문제, 같은 방법을 선택했더라도 코드가 다른 게 정말 신기하다. 나도 좀 더 간결하고 알아보기 쉽고 효율적인 코드를 짜고 싶다.