[프로그래머스] 숫자 문자열과 영단어 (python)

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

---

 

입출력 예	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초

---

🗒 풀이

def solution(s):
    units = {'zero': 0,
             'one': 1,
             'two': 2,
             'three': 3,
             'four': 4,
             'five': 5,
             'six': 6,
             'seven': 7,
             'eight': 8,
             'nine': 9}
    for key, value in units.items():
        s = s.replace(key, str(value))
    return int(s)

units이라는 영단어와 숫자를 저장하고 있는 변수를 만든다. 그다음, units.items()로 각 dictionary의 항목을 탐색한다.

이때, 키와 값 쌍으로 탐색하는 items()로 돌렸기 때문에 for문에 key, value를 각각 받을 수 있게 해줘야 한다. 만약 units의 zero, one, two 등 key값으로 돌리고 싶다면 units.keys(), 혹은 0, 1, 2 등과 같이 value값으로 돌리고 싶다면 units.values()로 실행한다.

만약, 입력받은 문자열 s에 key값, 즉 영문자가 있다면 replace 함수를 사용하여 해당 영문자의 값으로 바꿔준다.

이때, replace 한 값을 새로 저장해주지 않는다면, 제대로 반영이 되지 않으므로 문자열 s에 저장함으로써 값을 갱신해준다.

마지막으로 반환되는 값은 문자열이 아닌 정수 값이 여야 하므로 int(s)로 문자열을 정수 값으로 변환해준다.

---

🤔 후기

보자마자 dictionary를 사용해서 풀고 싶다는 생각은 들었다.

그리고 코드를 다음과 같이 짰다.

for key in units.keys():
    if key in s or str(units[key]) in s:
        answer += str(units[key])

일단 테스트 코드가 다 돌아가서 성공한 줄 알았으나, 간과한 것이 있었다.

테스트 코드는 숫자들이 섞인 것처럼 보이나, 사실은 오름차순의 테스트 케이스들만 있었다. dictionary units의 key값을 가지고 순회하기 때문에 "nine97four"과 같이 큰 값과 작은 값이 섞여있으면 제대로 된 값이 반환되지 않는 것이었다.

그래서 다시 생각해본 결과, 숫자 값은 잘 들어가 있는 거니 건드리지 않아도 되었고, 영문자만 처리하면 되는 것이었다.

비슷하게 s에 key, 즉 문자열이 있으면 해당 문자열을 숫자로 바꿔주면 되는 것이니 간단하게 처리할 수 있었다.

이때, dictionary의 value값들은 문자열이 아닌 정수열이다. 우리가 처리하는 s는 문자열이므로, str(units[key]) 처리를 해주지 않는다면 다음과 같은 오류가 발생할 수 있으니 주의하자.

TypeError: 'in <string>' requires string as left operand, not int

---

성공