본문 바로가기
Dev/Algorithm

[백준] #6443 애너그램 - 파이썬

by jusep 2025. 8. 11.

문제

씬디는 애너그램(anagram) 프로그램을 만들어 줄 수 있는 남자를 좋아한다. 참고로 씬디는 매우 예쁘다.

애너그램 프로그램이란, 입력받은 영단어의 철자들로 만들 수 있는 모든 단어를 출력하는 것이다. 가령 "abc" 를 입력받았다면, "abc", "acb", "bac", "bca", "cab", "cba" 를 출력해야 한다.

입력받은 단어내에 몇몇 철자가 중복될 수 있다. 이 경우 같은 단어가 여러 번 만들어 질 수 있는데, 한 번만 출력해야 한다. 또한 출력할 때에 알파벳 순서로 출력해야 한다.

입력

첫째 줄에 단어의 개수 N 이, 둘째 줄부터 N개의 영단어가 들어온다. 영단어는 소문자로 이루어져 있다. 단어의 길이는 20보다 작거나 같고, 애너그램의 수가 100,000개 이하인 단어만 입력으로 주어진다.

출력

각각의 영단어마다 모든 가능한 애너그램을 출력한다. 각각의 영단어에 대한 애너그램을 출력할 때, 알파벳 순서로 중복되지 않게 출력한다.

예제 입력 1 

2
abc
acba

예제 출력 1 

abc
acb
bac
bca
cab
cba
aabc
aacb
abac
abca
acab
acba
baac
baca
bcaa
caab
caba
cbaa

복기

1. 백트래킹과 dfs는 엄밀히 말하면 차이가 있다. dfs는 모든 구간을 탐색하지만 백트래킹은 그렇지 않다. 대신에 재귀적 성질을 가지는 것은 공통점이다.

2. anwer는 back_traking 외부에서 선언된 변수이지만, 파이썬에서 리스트는 가변 객체로 수정하면 유지가 된다. 

3. 중복은 dict를 통해서 방지한다.


솔루션

from collections import defaultdict

N = int(input())

def back_traking(cnt):
    if cnt == len(word):
        print("".join(answer))
        return

    #visited에 단어를 확인
    for k in visited:
        if visited[k]:
            visited[k] -= 1 # k를 사용할 것이므로 -1
            answer.append(k)
            back_traking(cnt+1) # 백트래킹
            visited[k] += 1
            answer.pop() # answer에서 빼준다.


for _ in range(N):
    word = (list(sorted(input().strip())))
    visited = defaultdict(int)
    answer = []
    for i in word:
        if i in visited:
            visited[i] += 1
        else:
            visited[i] = 1
    back_traking(0)

댓글