[프로그래머스 / DFS&BFS] 단어 변환 (java) ⭐️

문제 설명

두 개의 단어 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 합니다.

입출력 예

begintargetwordsreturn

hit	cog	[hot, dot, dog, lot, log, cog]	4
hit	cog	[hot, dot, dog, lot, log]	0

입출력 예 설명

예제 #1
문제에 나온 예와 같습니다.

예제 #2
target인 cog는 words 안에 없기 때문에 변환할 수 없습니다.

 

[문제접근]

각 단어를 그래프로 생각한다. 

목록에 타겟이 없으면 0을 리턴.

같은 단어가 두개 이상 있는 것들을 간선으로 연결한다. -> 그래프 구현

(그래프를 구현할 시 cycle이 생기지 않도록 주의한다.)

그래프 구현이 완성되면 BFS를 통해 타겟으로 갈때까지의 count를 센다.(최소 간선을 구해야 하므로 BFS를 사용한다.)

 

[java]

# 1차시도 - 실패

import java.util.*;
import java.io.*;

class Solution {
    static class Node{
        String word;
        int next;
        
        public Node(String word, int next){
            this.word = word;
            this.next = next;
        }
    }
    public int solution(String begin, String target, String[] words) {
        int answer = 0;
        
        makeGraph(begin, words);
        
        for(int i=0; i<words.length; i++){
                
        }
        return answer;
    }
    static void makeGraph(String begin, String[] words){
        int n = begin.length(); // 모든 단어의 길이는 같으므로 
        int count =0;
        Queue<Node> queue = new LinkedList<Node>();
        
        for(int i=0; i<words.length; i++){
            for(int j=0; j<n; j++){
                // 단어 비교
                if(begin.charAt(j) != words[i].charAt(j)) 
                    count++;
            }
            if(count == 1){
                
            }
        }
    }
}

무언가 이상하다는 것을 깨달았다. 

makeGraph로 그래프를 만든 후 bfs를 하면, 두번 탐색하게 되므로 시간이 낭비된다. 

그래서 맞는지 체크를 하면서 동시에 큐에 넣어주면서 체크를 하는 방식으로 해야한다. 

 

#2차시도

(다른 사람의 풀이를 참고했다.)

makeGraph 함수 대신 isNext라는 함수를 만들었다. 

import java.util.*;
import java.io.*;

class Solution {
    static class Node{
        String word;
        int edge; // 몇번째 간선
        
        public Node(String word, int edge){
            this.word = word;
            this.edge = edge;
        }
    }
    public int solution(String begin, String target, String[] words) {
        int answer = 0;
        
        Queue<Node> queue = new LinkedList<>();
        boolean[] visited = new boolean[words.length];
        
        queue.add(new Node(begin, 0));
        
        while(!queue.isEmpty()){
            Node cur = queue.poll();
            if(cur.word.equals(target)){
                answer = cur.edge;
                break;
            }
            
            for(int i=0; i<words.length; i++){
                if(!visited[i] && isNext(cur.word, words[i])){
                    visited[i] = true;
                    queue.add(new Node(words[i], cur.edge + 1));
                }
            }
        }
        
        return answer;
    }
    
    static boolean isNext(String cur, String next){
        int count = 0;
        for(int i=0; i<cur.length(); i++){
            if(cur.charAt(i) != next.charAt(i))
                count++;
            if(count > 1) 
                return false;
        }
        return true;
    }
    
}

 

 

 

[링크]

https://programmers.co.kr/learn/courses/30/parts/12421

프로그래머스