[프로그래머스 - 2021 카카오 채용연계형 인턴십] - 숫자 문자열과 영단어 (Python3)

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

---

입출력 예sresult

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

---

입출력 예 설명

입출력 예 #1

• 문제 예시와 같습니다.

입출력 예 #2

• 문제 예시와 같습니다.

입출력 예 #3

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

입출력 예 #4

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

---

제한시간 안내

• 정확성 테스트 : 10초

 

 

풀이1 (실패)

처음에는 이렇게 해도 되나? 싶은 생각으로 모두 if문 처리를 했었다. 인덱스가 s문자열의 길이와 같아질때까지 while문을 돌면서 문자가 z(0), o(1), t(2,3), f(4,5), s(6,7), e(8), n(9)인 경우에 맞게 인덱스를 증가시키고 해당하는 숫자를 결과 문자열에 넣어주는 방식으로 코드를 짰다.

 

하지만 시간 초과가 나왔고, 다른 방법을 생각해보았다.

 

풀이2 (성공)

def solution(s):
    answer = ''
    temp = ''
    str_num = ["zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine"]
    for i in range(len(s)):
        if s[i].isdigit():  # 숫자면 바로 넣어준다.
            answer += s[i]
        else:
            temp += s[i]
            if temp in str_num:   #### keys() : 딕셔너리의 key들의 리스트 반환
                answer += str(str_num.index(temp))
                temp = ''
    answer = int(answer)
    return answer

먼저 0~9까지의 영문이 들어 있는 리스트 str_num을 만들어주고

문자열 s를 탐색한다

 

isdigit()함수를 통해 문자가 숫자인지를 판단하고

문자가 숫자일 경우엔 그 숫자를 그대로 결과 문자열에 추가해주고, 숫자가 아니라면 temp 문자열에 넣어준다. temp값이 str_num에 들어 있는 값이 될 때까지 이를 반복하고, str_num에 매칭되는 문자열이 완성되면 str_num에서의 그 문자열의 인덱스를 결과 문자열에 추가해준다.

0~9까지의 숫자이기 때문에 마침 인덱스에 딱딱 맞게 매칭을 할 수 있었다. 이게 아니라면 딕셔너리 자료형을 사용해서 key:value의 쌍으로도 매칭시킬 수 있다.

참고로 딕셔너리 자료형의 .keys() 메서드를 사용하면 해당 딕셔너리의 key값들의 리스트를 받아올 수 있다.

 

 

문제를 풀고 나서 다른 사람의 풀이도 봤는데 replace를 사용해서 훨씬 간단하게 코드를 짤 수도 있었다.

그 풀이를 참고해서 내 코드를 다시 짜 보았다.

 

풀이3 (replace 사용)

def solution(s):
    answer = s
    str_num = ["zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine"]
    for i in range(len(str_num)):
        answer = answer.replace(str_num[i], str(i))
    return int(answer)

코드가 훨씬 간결해졌다.

 

 

유용한 함수

str.isdigit()

-> str문자열이 숫자인지 아닌지 판별

list.index(A)

-> list리스트의 원소 A의 인덱스를 반환

dict.keys()

-> dict딕셔너리의 key값들의 리스트를 반환

 

@@@

str.replace(A,B)

-> str문자열의 A를 B로 바꿔준다.