来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/word-ladder
给定两个单词(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" 不在字典中,所以无法进行转换。
解体思路:看完题目之后我的第一思路是从endWord出发,每次从wordList中找到和endWord只相差一个字母的单词,并将它们加入到list中,然后遍历list,重复之前的动作,直到找到beginWord或者list为空,注意到在此过程中将已经加入到list中的单词从wordList中删除。但是该方法一直出现超时的错误,后来想到可以从beginWord和endWord两端开始找,直到两个list中出现相同的单词或者某一个list为空。AC代码如下:
class Solution: def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int: if not (endWord in wordList): return 0 if len(beginWord) == 1: return 2 word_list = wordList[:] ans = 1 # 初始化答案 temp1 = self.cal(beginWord, wordList) ans += 1 # 从beginWord开始找 if self.result(temp1, endWord): # 判断是否结束 return ans temp2 = self.cal(endWord, word_list) ans += 1 while True: if self.result(temp1, temp2): return ans length1 = len(temp1) length2 = len(temp2) for i in range(length1): word = temp1.pop(0) temp1 += self.cal(word, wordList) ans += 1 if self.result(temp1, temp2): return ans for i in range(length2): word = temp2.pop(0) temp2 += self.cal(word, word_list) ans += 1 if not length1 or not length2: break return 0 def result(self, temp1, temp2): # 判断是够结束 for i in temp1: if i in temp2: return True return False def cal(self, word, word_list): # 计算和Word只差一个字母的单词 temp = [] visited = [] for i in word_list: if self.judge(i, word): temp.append(i) visited.append(i) # word_list.remove(i) for i in visited: word_list.remove(i) # print(word_list) return temp def judge(self, word1, word2): flag = 0 for i in range(len(word1)): if word1[i] != word2[i]: flag += 1 if flag == 1: return True else: return False