알고리즘) 프로그래머스 DFS/BFS, 단어 변환

프로그래머스 고득점 Kit - DFS/BFS

단어 변환

문제 설명

두 개의 단어 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

 

입출력 예 설명

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

 

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

 

풀이

가장 먼저 생각한 것은 Target으로 변환할 수 있는가였다. 이에 containTarget을 통해 Target으로 변환가능한지 여부를 따져 answer를 나누었다. 변환이 가능하다면, DFS를 반복하면서 한글자 차이가 나는 단어를 찾아 바꿔주면서 답을 찾았다. Target까지 완성이 되면 answer와 변경 횟수의 최소값을 갱신하면서, 비교적 쉬운 로직으로 풀었다. visit을 이용한 기본적인 dfs 문제라는 생각이 들었지만, 풀 때 고민을 조금 했던것 같다.

 + answer를 50으로 초기화한 이유는 단어의 총 개수가 50개여서 이를 넘어갈 수 없다고 생각했다.

 

코드

package programmers;

public class KitDBFSWordConverting {
    static int answer;
    private static int solution(String begin, String target, String[] words) {
        answer = 50;    // 총 단어가 50개
        int[] visit = new int[words.length];

        if(containTarget(words, target)) {  // 변환 가능하면
            dfs(visit, begin, target, words, 0);
        } else {
            answer =0;
        }

        return answer;
    }

    private static void dfs(int[] visit, String begin, String target, String[] words, int cnt) {
        if(begin.equals(target)) {  // 같은 글자가 나오게 되면
            answer = answer > cnt ? cnt : answer;   // 최솟값 저장
            return;
        }

        for(int i=0; i<words.length; i++) {
            if(visit[i] ==0) {  // 변경한 적이 없는 단어면
                if (canConvert(begin, words[i])) {
                    visit[i] = 1;
                    dfs(visit, words[i], target, words, cnt+1); // 변경
                    visit[i] = 0;   // 변경안하는 경우
                }
            }
        }
    }

    private static boolean canConvert(String origin, String target) {   // 한글자 차이나는지 확인
        int cnt =0;

        for(int i=0; i<origin.length(); i++) {
            if(origin.charAt(i) == target.charAt(i)) {
                cnt++;
            }
        }

        return cnt >= origin.length()-1 ? true : false;
    }

    private static boolean containTarget(String[] words, String target) {   // target이 단어 리스트에 있는지 확인
        for(String word : words) {
            if(word.equals(target)) {
                return true;
            }
        }
        return false;
    }

    public static void main(String[] args) {
        String begin = "hit";
        String target = "cog";
        String[] words = {"hot", "dot", "dog", "lot", "log", "cog"};
//        String[] words = {"hot", "dot", "dog", "lot", "log"};

        System.out.println(solution(begin,target,words));
    }
}