[프로그래머스] LV1. 숫자 문자열과 영단어 (파이썬, Python)

 

 

 

구분

코딩테스트 연습 > 2021 카카오 채용연계형 인턴십

 

문제 설명

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

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

• 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로 주어집니다.

---

입출력 예

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

---

입출력 예 #1

• 문제 예시와 같습니다.

입출력 예 #2

• 문제 예시와 같습니다.

입출력 예 #3

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

입출력 예 #4

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

---

제한시간 안내

• 정확성 테스트 : 10초

 

 

 

 

문제 풀이

def solution(s):
    # 어떻게 split할 것이냐
    # 숫자면 숫자 그대로 어떻게?
    dict = {"zero":"0", "one":"1", "two":"2", "three":"3", "four":"4", "five":"5", "six":"6", "seven":"7", "eight":"8", "nine":"9"}
    for key, value in dict.items():
        # 키를 값으로 바꾸기, 문자열만 가능
        s = s.replace(key, value)
    return int(s)

 

일단 문제를 보자마자 딕셔너리가 떠올랐다. 그러나 split()을 어떻게 해야하지? 라고 처음엔 막막했었다.

dict.items()를 통해 키&값을 꺼내고, replace()로 키에 해당되는 zero -> 0으로 바꾸는 작업을 했고, 마지막에 int()로 바꾸니 쉽게 풀리는 문제였다. 

 

 

만약 실전에서 dict.items()가 떠오르지 않는다면?

리스트와 인덱스를 활용해서도 문제를 풀 수 있었다. 

또한, isdigit() (문자열이 숫자로만 이루어져 있는지 여부를 확인하는 메서드) 를 기억해도 좋을 것 같다.

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)