1759번: 암호 만들기

https://www.acmicpc.net/problem/1759

---

l, c = map(int, input().split())
arr = list(map(str, input().split()))
arr.sort()

choose = []
check = ['a', 'e', 'i', 'o', 'u']

def combination(index, level):
  if level == l:
    cnt, cnt2 = 0, 0
    for ch in choose:
      if ch in check:
        cnt += 1
      else:
        cnt2 += 1

    if cnt >= 1 and cnt2 >= 2:
      print(''.join(choose))

    return
  
  for i in range(index, c):
    choose.append(arr[i])
    combination(i + 1, level + 1)
    choose.pop()

combination(0, 0)

 

생각

최소 한 개의 모음 + 최소 두 개의 자음, 사전 순으로 나열해야 하는 조합 문제

1. 문자열을 리스트 형식으로 바꾼 후 정렬해준다.

2. check 배열에 모음을 담아준다.

3. 재귀 함수를 이용하여 조합 함수를 만들어준다.

4. 모음인 문자는 cnt + 1, 모음이 아니면 자음이므로 cnt2 + 1을 해준다.

5. 최소 한 개의 모음(cnt >= 1) + 최소 두 개의 자음(cnt2 >= 2)에 해당하는 문자열을 출력해준다.