programmers

[Programmers]깊이/너비 우선 탐색(DFS/BFS)>단어 변환

윤만석 2022. 8. 26. 11:16

문제 설명

두 개의 단어 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 합니다.
#include <string>
#include <vector>
#include<queue>
#include<cstring>
using namespace std;

int visited[51];
int dif(string &a,string &b){
    int d=0;
    for(int i=0;i<a.length();i++){
        if(a[i]!=b[i]){
            d++;
        }
    }
    return d;
}
int solution(string begin, string target, vector<string> words) {
    memset(visited,-1,sizeof(visited));
    words.push_back(begin);
    queue<int>q;
    q.push(words.size()-1);
    visited[words.size()-1]=0;
    while(!q.empty()){
        int top=q.front();
        q.pop();
        if(words[top]==target)return visited[top];
        for(int i=0;i<words.size();i++){
            if(top==i)continue;
            if(visited[i]== -1 && dif(words[top],words[i])==1){
                q.push(i);
                visited[i]=visited[top]+1;                
            }
        }
    }
    return 0;
}

프로그래머스 leve3 dfs/bfs 문제입니다.

begin으로부터 target으로 단어 하나씩 변화시키면서 바꾸면 됩니다.

target단어는 words에 있는데 begin은 없어서 words에 begin을 push해주고

words 마지막 index로 시작해서 bfs를 실행합니다.

함수 dif는 두 단어의 차이로, for문을 돌려가면서 방문하지 않은경우 (visited=-1) 그리고 dif=1인 경우 queue에 push합니다. 그리고 방문하면 visited[i]=visited[index]+1로 저장합니다.