쉬운 프로그래밍

[알고리즘] 프로그래머스 - 단어 변환 - JAVA 본문

알고리즘/백트래킹

[알고리즘] 프로그래머스 - 단어 변환 - JAVA

쉬운형 2021. 8. 12. 17:26

문제 설명

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

입출력 예

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

입출력 예 설명

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

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

 


풀이 과정

입력으로 주어진 begin을 target으로 변환하는 최소비용을 찾는 문제다. 

 

변환 규칙은 begin을 시작으로 words 배열 내에 있는 문자열들 중에서 현재 문자열과 한글자만 다른 문자열로 변환할 수 있다.

 

모든 경우를 탐색해야하고, 방문 순서에 따라 비용이 달라지기 때문에 백트래킹을 통해 문제를 풀었다.

 

문자열이 한글자만 다른지 판별하는 check 메소드를 만들어 주었으며, 현재 시점에 방문하지 않은 정점이며 check 메소드를 통과하는 경우만 DFS를 호출하도록 구현했다.

 

종료 조건은 begin과 target이 같은 경우며, 현재 비용이 이전에 구하였던 비용보다 작은지 판별한다.

 

소스코드

class Solution {
    static boolean[] visited;
    static int value = 51;
    
    public int solution(String begin, String target, String[] words) {
        visited = new boolean[words.length];
        backTracking(begin, target, words, 0);
        
        if (value == 51) {
            return 0;
        }
        
        return value;
    }
    
    public void backTracking(String begin, String target, String[] words, int depth) {
        if (begin.equals(target)) {
            value = Math.min(value, depth);
            return;
        }
        for (int i = 0; i < words.length; i++) {
            if (!visited[i] && check(begin, words[i])) {
                visited[i] = true;
                backTracking(words[i], target, words, depth + 1);
                visited[i] = false;
            }
            
        }
    }
    
    public boolean check(String begin, String target) {
        int count = 0;
        
        for (int i = 0; i < begin.length(); i++) {
            if (begin.charAt(i) == target.charAt(i)) {
                count++;
            }    
        }
        
        if (count == begin.length() - 1) {
            return true;
        }
        return false;
    }
}

 

 

Comments