오늘은 백트래킹에 대해 학습했습니다.
요즘 계속해서 알고리즘/자료구조를 공부하고있는데 DFS/BFS 개념이 나온뒤로 이해가 안되고 끙끙 앓고있다가 재귀함수의 흐름에 대해 감히 잡히더니 문제를 못풀어도 다른 사람들이 푼 문제를 보면 이해가 가는 수준이 되었습니다.... 노력하면 안될건 없다는 말이 생각나네요 허허
백트래킹이란 현재 노드에서 다음 노드로 가는 모든 경우의 수를 찾은 후 필요 없는 노드를 가지치기 함으로써 시간복잡도를 줄이는 방법
문제
바로 어제 최백준 조교가 방 열쇠를 주머니에 넣은 채 깜빡하고 서울로 가 버리는 황당한 상황에 직면한 조교들은, 702호에 새로운 보안 시스템을 설치하기로 하였다. 이 보안 시스템은 열쇠가 아닌 암호로 동작하게 되어 있는 시스템이다.
암호는 서로 다른 L개의 알파벳 소문자들로 구성되며 최소 한 개의 모음(a, e, i, o, u)과 최소 두 개의 자음으로 구성되어 있다고 알려져 있다. 또한 정렬된 문자열을 선호하는 조교들의 성향으로 미루어 보아 암호를 이루는 알파벳이 암호에서 증가하는 순서로 배열되었을 것이라고 추측된다. 즉, abc는 가능성이 있는 암호이지만 bac는 그렇지 않다.
새 보안 시스템에서 조교들이 암호로 사용했을 법한 문자의 종류는 C가지가 있다고 한다. 이 알파벳을 입수한 민식, 영식 형제는 조교들의 방에 침투하기 위해 암호를 추측해 보려고 한다. C개의 문자들이 모두 주어졌을 때, 가능성 있는 암호들을 모두 구하는 프로그램을 작성하시오.
입력
첫째 줄에 두 정수 L, C가 주어진다. (3 ≤ L ≤ C ≤ 15) 다음 줄에는 C개의 문자들이 공백으로 구분되어 주어진다. 주어지는 문자들은 알파벳 소문자이며, 중복되는 것은 없다.
출력
각 줄에 하나씩, 사전식으로 가능성 있는 암호를 모두 출력한다.
import sys
def dfs(index , path):
if len(path) == L :
vol = 0
jel = 0
for i in range(len(path)):
if path[i] in vowel:
vol +=1
else:
jel +=1
if vol >=1 and jel >=2:
print("".join(path))
return
for i in range(index, len(A)):
dfs(i+1,path+A[i])
vowel = ['a','e','i','o','u']
L,C = map(int,sys.stdin.readline().split())
A = sorted(list(map(str,sys.stdin.readline().split())))
#list = []
#for i in A:
#list.append(i)
#list.sort()
dfs(0,"")
처음에
if len(path) == L :
vol = 0
jel = 0
for i in range(len(path)):
if path[i] in vowel:
vol +=1
else:
jel +=1
if vol >=1 and jel >=2:
print("".join(path))
return
이렇게 말고
if len(path) == L :
if ('a' 'e' 'i' 'o' 'u') in not path:
path.sort()
path.sort()
result.append(path)
를 했었는데 그때는 자음이 2개 이상 들어가야한다는 조건을 생각을 못했음. 이렇게 실행했는데 cstw가 나오면 안되는데 계속 수정을 해봐도 나오길래 다른사람의 코드를 비교했음.
원래 다른 사람의 코드를 봐도 재귀함수가 실행되는 흐름이 이해가 안가서 코드를 이해하지 못했는데 이제는 이해할 수 있게 됨.
다른 사람의 코드를 확인하고
#list = []
#for i in A:
#list.append(i)
#list.sort()
위 4줄로 코드를 짰었는데 pycharm에서 실행하면 결과가 잘 나오다가도 백준에 제출하면 런타임 에러가 발생함 그래서 코드를
A = sorted(list(map(str,sys.stdin.readline().split())))
for문을 사용 안하고 바로 한줄로 코드를 깔끔하게 수정하니까 런타임에러가 발생하지 않음.
여태까지 깔끔하게 코드 짤 수 있던것도 주저리주저리 길게 코드를 짰었는데 이번 기회에 깔끔하게 코드를 짤 수 있도록 고민해야겠다는 생각이 들었음.