[파이썬][프로그래머스] 모음 사전
1. 문제Permalink
[level 2] 모음 사전 - 84512Permalink
성능 요약Permalink
메모리: 10.3 MB, 시간: 0.01 ms
구분Permalink
코딩테스트 연습 > 완전탐색
채점결과Permalink
정확성: 100.0
합계: 100.0 / 100.0
문제 설명Permalink
사전에 알파벳 모음 'A', 'E', 'I', 'O', 'U'만을 사용하여 만들 수 있는, 길이 5 이하의 모든 단어가 수록되어 있습니다. 사전에서 첫 번째 단어는 "A"이고, 그다음은 "AA"이며, 마지막 단어는 "UUUUU"입니다.
단어 하나 word가 매개변수로 주어질 때, 이 단어가 사전에서 몇 번째 단어인지 return 하도록 solution 함수를 완성해주세요.
제한사항
- word의 길이는 1 이상 5 이하입니다.
- word는 알파벳 대문자 'A', 'E', 'I', 'O', 'U'로만 이루어져 있습니다.
입출력 예
word | result |
---|---|
"AAAAE" |
6 |
"AAAE" |
10 |
"I" |
1563 |
"EIO" |
1189 |
입출력 예 설명
입출력 예 #1
사전에서 첫 번째 단어는 "A"이고, 그다음은 "AA", "AAA", "AAAA", "AAAAA", "AAAAE", ... 와 같습니다. "AAAAE"는 사전에서 6번째 단어입니다.
입출력 예 #2
"AAAE"는 "A", "AA", "AAA", "AAAA", "AAAAA", "AAAAE", "AAAAI", "AAAAO", "AAAAU"의 다음인 10번째 단어입니다.
입출력 예 #3
"I"는 1563번째 단어입니다.
입출력 예 #4
"EIO"는 1189번째 단어입니다.
출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges
2. 해결방법 시간복잡도Permalink
- 등비수열 O(N)
3. 문제 해결 및 코드Permalink
def solution(word): | |
answer = 0 | |
for i, n in enumerate(word): | |
answer += (5 ** (5 - i) - 1) / (5 - 1) * "AEIOU".index(n) + 1 | |
# 문자의 위치만큼 5의 제곱만큼 계산후에 -1을 해준다. | |
# 5개 문자로 만들 수 있는 최대 숫자의 길이는 4이기 때문이기에 4로 나눠준다. | |
# 이후에 AEIOU에서 해당 값을 검색한 뒤 + 1을 더해준다. | |
return answer |
-
주석을 참고하면서 이해를 돕습니다.Permalink
4. 알고리즘 및 해설Permalink
- 반복문을 통해서 문제를 해결해준다.
- 이때 문제에서 나오는 값을 기반으로 5개의 문자이므로 5의 제곱 * 이후 해당 위치만큼 빼주는 데 이때 5개의 문자로 만들 수 있는 문자열의 길이는 4이므로 1을 빼주고 나눠준다.
- 그리고 AEIOU라는 문장에서 해당 문자가 어디에 위치하는 지 찾아준 값을 곱해준 뒤 최종적으로 1을 더해준다.
- 최종 값을 출력해준다.