본문 바로가기
개인 공부/코딩테스트

[C++][프로그래머스][BFS DFS] 단어변환 :: seoftware

by seowit 2020. 3. 1.

문제 설명

두 개의 단어 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] 0
hit cog [hot, dot, dog, lot, log, cog] 4

입출력 예 설명

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

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

소스코드

#include <string>
#include <vector>
#include <queue>

using namespace std;
queue<int> q;
int result[50];
bool visited[50];

int solution(string begin, string target, vector<string> words) {
    int answer = 0;
    int len = begin.length();
    for(int i = 0; i < words.size(); i++) {
        int cnt = 0;
        for(int j = 0; j<len; j++){
            if(begin[j] != words[i][j]) cnt++;
            if(cnt == 2) break;
        }
        if(cnt == 1) {
            q.push(i); visited[i] = true;
            result[i] = 1;
        }
    }
    while(!q.empty()){
        int now_idx = q.front();
        string now = words[now_idx];
        q.pop();
        //종료조건
        for(int i = 0; i < len; i++){
            if(target[i] != now[i]) break;
            if(i == len-1) return result[now_idx];
        }
        //추가
        for(int i = 0; i < words.size(); i++) {
            int cnt = 0;
            if(!visited[i]){
                for(int j = 0; j < len; j++){
                    if(now[j] != words[i][j]) cnt++;
                    if(cnt == 2) break;
                }
                if(cnt == 1) {
                    q.push(i);
                    visited[i] = true;
                    result[i] = result[now_idx]+1;
                }
            }
            else continue;
        }
    }
    return answer;
}

1. begin과 한글자만 다른 단어의 인덱스를 큐에 넣는다. (visited는 true로 바꾸고 result는 +1 해줌)

2. while문 시작. 큐의 맨 앞 단어가 target이면 종료한다. 지금보니까 for문으로 쓰지말고 now == target 으로 해도 되는걸 괜히 저렇게 했다.

3. target이 아닌 경우에는 방문하지 않은 단어 중 한 글자만 다른 것을 큐에 넣고 result에 이전꺼(자기가 파생된 단어)에 +1 해준다.

4. 만약 중간에 종료조건에 의해 끝나지 않는 것은 target으로 변환에 실패했다는 것이고 answer(0)을 반환한다.

댓글