[백준/Python] 6443번 애너그램

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

6443번: 애너그램

---

문제

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

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

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

 

입력

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

 

출력

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

 

코드

import sys
ssr = sys.stdin.readline


def backT(word, idx):
    if len(word) == len(words[idx]):
        print(''.join(word))
        return

    for k in wordD[idx]:
        if wordD[idx][k]:
            word.append(k)
            wordD[idx][k] -= 1

            backT(word, idx)

            word.pop()
            wordD[idx][k] += 1


N = int(ssr())
wordD = []
words = []

for i in range(N):
    inWord = sorted(ssr().strip())
    words.append(inWord)
    visited = {}

    for j in inWord:
        if j in visited:
            visited[j] += 1
        else:
            visited[j] = 1

    wordD.append(visited)


for i in range(N):
    backT([], i)

 

 

문제풀이

1. 입력한 영단어에 대한 "모든" 가능한 단어를 출력해야 하기 때문에 백트래킹을 이용한다.

2. "abc"를 입력 시 "aaa"와 같은 중복 사용 방지를 위해 사용된 알파벳이 재사용됐는지 확인해야 한다. (일명 visited를 사용한다.)

3. visited는 딕셔너리로 구현. key 값은 알파벳, value 값은 알파벳의 사용 가능한 횟수로 구현.

막혔던 부분

최초에 visited를 구현할 때 2중 배열을 이용하여 [사용 -> True, 미사용 -> False]로 구현하였고, 중복 방지를 위해 단어들을 리스트에 넣어 [not in]을 이용하여 체크하였다. 하지만 위와 같이 구현하니 [시간 초과] 혹은 [메모리 초과]가 발생하였다.때문에 위 코드와 같이 visited를 딕셔너리로 구현하여 key를 알파벳으로, value를 사용 가능한 횟수로 설정하여 해결하였다.