[baekjoon] 백준 1759번(파이썬): 암호 만들기

문제

1759번: 암호 만들기

알고리즘

- 백 트래킹을 통해 문제를 수행한다.

- 암호를 만들었다면 암호에 모음과 자음의 개수를 체크한다.

- 모음이 1개 이상, 자음이 2개 이상이라면 암호를 잘 만든 것으로 출력한다.

- 암호는 반복문과 백 트래킹을 통해 만든다.

코드

import sys


def back_tracking(cnt, idx):

    # 암호를 만들었을 때
    if cnt == l:
        # 모음, 자음 체크
        vo, co = 0, 0

        for i in range(l):
            if answer[i] in consonant:
                vo += 1
            else:
                co += 1

        # 모음 1개 이상, 자음 2개 이상
        if vo >= 1 and co >= 2:
            print("".join(answer))

        return

    # 반복문을 통해 암호를 만든다.
    for i in range(idx, c):
        answer.append(words[i])
        back_tracking(cnt + 1, i + 1) # 백트래킹
        answer.pop()


l, c = map(int, sys.stdin.readline().split())
words = sorted(list(map(str, sys.stdin.readline().split())))
consonant = ['a', 'e', 'i', 'o', 'u']
answer = []
back_tracking(0, 0)

github

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법