2021 카카오 채용연계형 인턴십 > 숫자 문자열과 영단어

문제 설명

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

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

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

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

---

내 풀이

def solution(s):
    if s.isdigit():
        return int(s)
    answer = s
    alpha = ['zero', 'one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine']
    
    for a in alpha:
        answer = answer.replace(a, str(alpha.index(a)))
    
    return int(answer)

통과 너무 조하

1. 일단, 입력된 데이터가 이미 숫자로 구성되어있으면 그대로 리턴한다 (int로 변환하여)

2. 숫자는 0-9까지로 리스트의 인덱스와 동일하게 사용할 수 있기 때문에 0-9를 의미하는 알파벳을 리스트로 구성했다.

3. alpha 리스트를 돌면서 입력받은 문자열에 숫자로 치환되어야 할 영단어가 있을 경우 인덱스로 치환한다

 

--> 알고리즘의 복잡도는 alpha리스트의 크기에 비례한다. alpha 크기는 10이니까 O(10) 임