백준 1759번 파이썬

• 문제링크 : https://www.acmicpc.net/problem/1759

 

구현

from itertools import combinations

mo_list = ['a', 'e', 'i', 'o', 'u']

def is_possible(word):
	global L, arr

	mo_cnt = 0
	for w in word:
		mo_cnt += (w in mo_list)
	ja_cnt = chose_num - mo_cnt

	return (mo_cnt >=1 and ja_cnt >= 2)

chose_num , total_num = map(int, input().split())
arr = input().split()
arr.sort()

for word in combinations(arr, chose_num):
	if is_possible(word):
		print(''.join(word))

• 조합 알고리즘을 사용함.
• combinations 함수를 이용함.