단어변환 | 프로그래머스 lv3

yoongyum·2022년 7월 5일
0

코딩테스트 🧩

목록 보기
43/47
post-thumbnail

🔎문제설명

두 개의 단어 begin, target과 단어의 집합 words가 있습니다. 아래와 같은 규칙을 이용하여 begin에서 target으로 변환하는 가장 짧은 변환 과정을 찾으려고 합니다.

1. 한 번에 한 개의 알파벳만 바꿀 수 있습니다.
2. words에 있는 단어로만 변환할 수 있습니다.
예를 들어 begin이 "hit", target가 "cog", words가 ["hot","dot","dog","lot","log","cog"]라면 
"hit" -> "hot" -> "dot" -> "dog" -> "cog"와 같이 4단계를 거쳐 변환할 수 있습니다.

두 개의 단어 begin, target과 단어의 집합 words가 매개변수로 주어질 때, 최소 몇 단계의 과정을 거쳐 begin을 target으로 변환할 수 있는지 return 하도록 solution 함수를 작성해주세요.

제한사항

  • 각 단어는 알파벳 소문자로만 이루어져 있습니다.
  • 각 단어의 길이는 3 이상 10 이하이며 모든 단어의 길이는 같습니다.
  • words에는 3개 이상 50개 이하의 단어가 있으며 중복되는 단어는 없습니다.
  • begin과 target은 같지 않습니다.
  • 변환할 수 없는 경우에는 0를 return 합니다.



🧊파이썬 코드

dfs 사용 코드

def solution(begin, target, words):
    global answer
    answer = 51
    if target not in words: return 0
    #words의 단어 중 알파벳이 한개의 차이가 있는 지 확인.
    def find(str1, str2):
        check = 0
        for s1, s2 in zip(str1, str2):
            if s1 != s2:
                check += 1
        return check == 1
    
    def dfs(words, ss, cnt):
        global answer
        if ss == target:
            answer = min(cnt, answer)
            return 
        for i in range(len(words)):
            if find(ss, words[i]):
                dfs(words[:i] + words[i+1:],words[i],cnt+1)
                
    dfs(words,begin,0);

    return answer

dfs로 리스트를 넘겨줄 때
words[:i] + words[i+1:] 이렇게 한 이유는
해당 words[i]는 한번 사용했기 때문에 제외하고 새로운 리스트로 만들어서 파라미터로 넘겨줍니다.

리스트 크기가 하나씩 줄면 점점 탐색하는 속도가 빨라져서 시간초과가 나지 않습니다.


bfs 사용 코드

from collections import deque

def solution(begin, target, words):
    #words의 단어 중 알파벳이 한개의 차이가 있는 지 확인.
    def find(str1, str2):
        check = 0
        for s1, s2 in zip(str1, str2):
            if s1 != s2:
                check += 1
        return check == 1
    
    Q = deque()
    Q.append([begin,[]])  #처음엔 비긴과 빈리스트를 넣음
    
    while Q:
        cur, save = Q.popleft()
        if cur == target:
            return len(save) #사용한 단어 수 리턴
        for word in words:
            if find(word,cur) and word not in save:
                tmp = save[:] #깊은 복사
                tmp.append(word)
                Q.append([word,tmp])
                     
    return 0

save는 사용한 단어들을 저장하는 리스트입니다.

보통 DFS에서 checked나 visited 리스트를 사용하는 것과 비슷합니다.

원래는 permutations함수를 사용해서 모든 순열을 구하려고 했지만,

조건에서 words의 최대 갯수가 50개라 시간초과가 발생합니다.

0개의 댓글