cje_ 2024. 5. 5. 17:31

https://school.programmers.co.kr/learn/courses/30/lessons/43163?language=go

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


문제 설명

두 개의 단어 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 안에 없기 때문에 변환할 수 없습니다.


풀이

bfs로 해결해야 하는 문제이다.

이 문제의 경우 begin에서부터 target이라는 단어로 변환하는 가장 짧은 과정을 찾아야 한다. 즉, 최단 경로와 흡사하다고 볼 수 있다.

첫번째 단어부터 변환하여 한 번 변환했을 때 다음 단어가 되면 true, 그게 아니면 false라 하면 되는 것이다. 그 과정이 calc 함수에 담겨 있다. 변환이 된다면 queue에 담고 다시 반복한다. 이러면서 target 값을 찾는다면 지금까지 단계를 반환해주면 된다.


코드

// 두 단어 비교
func calc(word1, word2 string) bool { // 매개변수 둘 다 string, 리턴 bool 
	diffCount := 0 // 비교 카운터 초기 값 0
	for i := range word1 { // word1을 순회하면서
		if word1[i] != word2[i] { // 같은 게 있다면
			diffCount++ // 카운터 증가
		}
		if diffCount > 1 { // 두 개 이상 다르면
			return false // false를 반환
		}
	}
	return diffCount == 1 // 한 글자만 다르면 true를 반환
}

func solution(begin string, target string, words []string) int {
	wordMap := make(map[string]bool) // 맵 생성
	for _, word := range words {
		wordMap[word] = true // 리스트를 맵으로 변환 
	}

	if !wordMap[target] {
		return 0 // target 없음 -> 변환 X
	}

	queue := []struct {	// 큐 초기화 (구조체 생성)
		word  string
		lv int
	}{ {word: begin, lv: 0} }
	visited := make(map[string]bool)
	visited[begin] = true // 시작 단어 -> 방문 처리

	// BFS
	for len(queue) > 0 { // 큐가 빌 때까지
		current := queue[0] // 하나 꺼내고
		queue = queue[1:] // 슬라이싱
		if current.word == target {// target 단어 도달
			return current.lv // 현재 단계 반환
		}

		for word := range wordMap { // 단어 검사
			if !visited[word] && calc(current.word, word) { // 방문 안 함 & 한 글자만 다름
				visited[word] = true // 방문 처리
				queue = append(queue, struct{word string; lv int}{word, current.lv + 1}) 
			} // 큐에 추가
		}
	}
	return 0
}