Leetcode练习(Python):广度优先搜索类:第127题:单词接龙:给定两个单词(beginWord 和 endWord)和一个字典,找到从 beginWord 到 endWord 的最短转换序列的长度。转换需遵循如下规则: 每次转换只能改变一个字母。 转换过程中的中间单词必须是字典中的单词。
题目:
单词接龙:给定两个单词(beginWord 和 endWord)和一个字典,找到从 beginWord 到 endWord 的最短转换序列的长度。转换需遵循如下规则: 每次转换只能改变一个字母。 转换过程中的中间单词必须是字典中的单词。
说明:
如果不存在这样的转换序列,返回 0。
所有单词具有相同的长度。
所有单词只由小写字母组成。
字典中不存在重复的单词。
你可以假设 beginWord 和 endWord 是非空的,且二者不相同。
示例 1:
输入:
beginWord = "hit",
endWord = "cog",
wordList = ["hot","dot","dog","lot","log","cog"]
输出: 5
解释: 一个最短转换序列是 "hit" -> "hot" -> "dot" -> "dog" -> "cog",
返回它的长度 5。
示例 2:
输入:
beginWord = "hit"
endWord = "cog"
wordList = ["hot","dot","dog","lot","log"]
输出: 0
解释: endWord "cog" 不在字典中,所以无法进行转换。
思路:
深度优先使用栈,广度优先使用队列。
程序:
from collections import deque class Solution: def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int: if endWord not in wordList: return 0 # 单词中缺少一个字母的单词组合 def make_dict(wordList): myDcit = {} for word in wordList: for index in range(len(word)): auxiliary = word[: index] + '_' + word[index + 1 :] myDcit[auxiliary] = myDcit.get(auxiliary, []) + [word] return myDcit # 广度优先 def bfs(beginWord, endWord, myDcit): # 初始设置为第一步 queue = deque([(beginWord, 1)]) # 保存访问过的单词 visited = set() while queue: word, step = queue.popleft() # 访问没有访问过的单词 if word not in visited: # 现在访问了 visited.add(word) # 判断是否得到结果 if word == endWord: return step for index in range(len(word)): # 按照现在的word进行查找 auxiliary = word[: index] + '_' + word[index + 1 :] neighbors = myDcit.get(auxiliary, []) for neighbor in neighbors: if neighbor not in visited: queue.append((neighbor, step + 1)) return 0 myDcit = make_dict(wordList) return bfs(beginWord, endWord, myDcit)