단어변환 | 프로그래머스 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개의 댓글

관련 채용 정보