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

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

 

입출력 예

 

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):
    answer = ''
    words = {
        "zero" : "0",
        "one" : "1",
        "two" : "2",
        "three" : "3",
        "four" : "4",
        "five" : "5",
        "six" : "6",
        "seven" : "7",
        "eight" : "8",
        "nine" : "9"
    }
    tmpS = ""
    for i in s:
        if i.isdigit():
            answer += i
            continue
        tmpS += i
        if tmpS in words:
            answer += words[tmpS]
            tmpS = ""
    
    return int(answer)

 

주어진 문자열에서 숫자는 숫자로, 영단어를 숫자로 바꾸어 문자열을 만들고, 마지막에 정수로 변환해 출력해주면 된다.

영단어를 비교해 찾는것은 그리 어렵지 않았지만, 숫자를 어떻게 비교할까 고민했다. 주어진 입력은 문자열이라서 어떻게 나누어야 할지 몰랐는데 파이썬의 isdigit() 함수를 사용했다.

 

isdigit

 

isdigit은 string 클래스에 있는 메서드로, 문자열.isdigit()의 형태로 사용할 수 있다.

 

문자열.isdigit()

 

문자열이 '숫자'로 이루어져 있는지 확인하는 함수이다.

문자가 하나라도 있다면 False, 모든 문자가 숫자로 이루어지면 True를 반환한다.

 

사용법

 

• str.isdigit("문자열")
• "문자열".isdigit()

 

문제 풀이로 돌아가면 주어진 s 문자열을 한 글자씩 탐색한다. i가 isdigit() 함수를 통해 True가 나오면 숫자라는 소리고, 그대로 answer에 더해준다. 그리고 숫자를 찾았으니 다음 원소를 비교하러 간다. 숫자가 아니라면 tmpS에 i를 더해주고, 더해진 tmpS가 words dictionary에 있다면 영단어로 취급해 해당 key에 맞는 value를 answer에 더해주고 tmpS를 초기화해준다. 마지막으로 s가 의미하는 원래 숫자를 return 하도록 int 형으로 형 변환해준다.