숫자 문자열과 영단어 (Level 1)

문제

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

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

 

내 풀이

 

1. table의 인덱스를 사용하려고 리스트 1개를 만들었다.
2. s의 원소가 숫자일 때는 바로 answer에 추가하고, 알파벳일 때는 원소를 임의의 변수에 저장한 다음 table에 있는 단어인지 확인하도록 하였다.

 

def solution(s):
    answer = ''
    table = ['zero', 'one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine']
    tmp, idx = '', 0
    for i in range(len(s)):
        if s[idx].isdigit(): answer += s[idx]
        if s[idx].isalpha():
            tmp += s[idx]
            if tmp in table:
                answer += str(table.index(tmp))
                tmp = ''
        idx += 1
        
    return int(answer)

 

다른 풀이

 

1. 영어와 숫자를 문자열 형태로 num_dic라는 dictionary에 저장하였다.
2. answer에서 key(영어) 부분을 전부 value(숫자)로 바꾸기 위해 문자열을 변경하는 함수인 replace 함수를 사용하였다.

 

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)

 

총평

replace 하나만으로 정말 간단하게 코드를 작성할 수 있는 것을 배웠다.

파이썬 함수는 양파같은 놈들이다. 까도 까도 새로운 게 나오는 느낌이랄까.