190325 DFS/BFS (3) 단어 찾기

2019-03-25

DFS/BFS (3) 단어 찾기

문제 설명

두 개의 단어 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 합니다.
입출력 예
begin target words return
hit cog [hot, dot, dog, lot, log, cog] 4
hit cog [hot, dot, dog, lot, log] 0
풀이 과정
  • BFS로 풀면 효율적인 유형에 약한 것 같다.
    • 코테전까지 DFS/BFS 한번 정복해보자.
  • begin부터 시작해서 words 배열을 BFS로 탐색
    • 만약 cur(현재 노드)의 word가 target으로 변환 가능하면, 현재 노드 단계를 리턴
    • cur에서 변환이 가능하고, 방문하지 않았던 노드라면(!visited[i]) 큐에 넣어 다음번 탐색에 이용
import java.util.*;
public class Node{
    String word;
    int level;
    public Node(String word,int level){
        this.word=word;
        this.level=level;
    }
}
class Solution {
    public boolean isContains(String[] words, String target){
        for(String w:words){
            if(w.equals(target)){
                return true;
            }
        }
        return false;
    }
    public boolean isChangable(String w1, String w2){
        int cnt=0;
        for(int i=0;i<w1.length();i++){
            if(w1.charAt(i)!=w2.charAt(i)){
                cnt++;
            }
        }
        return cnt==1;
    }
    public int solution(String begin, String target, String[] words) {
        if(!isContains(words,target)){
            return 0;
        }
        
        Queue<Node> q=new LinkedList<>();
        q.offer(new Node(begin,1));
        
        boolean[] visited=new boolean[words.length];
        while(!q.isEmpty()){
            Node cur=q.poll();
            if(isChangable(cur.word,target)){
                return cur.level;
            }
            for(int i=0;i<words.length;i++){
                if(!visited[i] && isChangable(cur.word, words[i])){
                    q.offer(new Node(words[i], cur.level+1));
                    visited[i]=true;
                }
            }
        }
        return 0;
    }
}