알고리즘
프로그래머스)모음사전-pyhton
1일1공부실천하자
2023. 2. 13. 01:01
https://school.programmers.co.kr/learn/courses/30/lessons/84512
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
from itertools import product
def solution(word):
arr = ["A","E",'I','O','U']
words = []
for i in range(1,6):
for c in product(arr,repeat=i):
words.append(''.join(list(c)))
words.sort()
idx = words.index(word)
return idx+1
처음엔 permutions로 모든 조합을 짠 뒤 인덱스를 출력하려고 했다.
하지만 ['A','A','A','A','A','E','E','E'......'U']
로 조합을 짰더니 첫번째인 'A'나 두번째인 'AA'는 생각하지 못했다.
결국 검색을 이용했고 , 데카르트곱을 출력하는 product를 이용했다.