Programmers 단어 변환 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 합니다.

 

나의 풀이

 가장 가까운거리 라는 점에서 BFS 방식을 써야겠다 생각을 하고 문제풀이를 시작. 주어진 단어에서 1글자만 다른 경우가 인접 노트라 보고 먼저 그래프를 그려봤다. 이 문제를 DFS로도 풀 수 있겠다 싶었지만, 괜한 반복 없이 답을 구하기엔 BFS가 적합하다 판단하고 코드를 구현했다. BFS에 대해 처음 익힐땐 각 노드마다 인접 노드를 저장하는 방식으로 연습했었는데, 이 문제는 인접노드를 저장할 필요는 없고 그냥 각 노드(단어) 마다 시작하는 단어에서 얼마나 떨어져 있는지와 방문했던 노드인지만 저장하면 됐다. 

 

코드

import java.util.*;

class Solution {
    class Node {
        String word;
        int leng;
        
        public Node(String word, int leng) {
            this.word = word;
            this.leng = leng;
        }
    }
    
    public int solution(String begin, String target, String[] words) {
        int answer = 0;
        Queue<Node> que = new LinkedList<>();
        boolean[] visited = new boolean[words.length];
        que.offer(new Node(begin, 0));
        
        while(!que.isEmpty()) {
            Node n = que.poll();
            
            if(n.word.equals(target)) {
                answer = n.leng;
                break;
            }
            
            for(int i=0; i<words.length; i++) {
                if(visited[i] || !checkAd(n.word, words[i])) {
                    continue;
                }
                visited[i] = true;
                que.offer(new Node(words[i], n.leng+1));
            }
        }
        
        return answer;
    }
    
    static boolean checkAd(String w1, String w2) {
        int count = 0;
        for(int i=0; i<w1.length(); i++) {
            if(w1.charAt(i)!=w2.charAt(i))
                count++;
            if(count>1) return false;
        }
        return true;
    }
}