赞
踩
给定两个单词(beginWord 和 endWord)和一个字典,找到从 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" 不在字典中,所以无法进行转换。
解题思路
很明显这是一个图的问题。对于示例1
来说
由于是求最短路径,我们很容易想到通过广度优先遍历来解决这个问题。现在我们要解决的问题就变成了如何判断两个单词只有一个字母不同。最简的办法就是通过26个字母替换
class Solution: def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int: wordDict = set(wordList) if endWord not in wordDict: return 0 q, visited = [(beginWord, 1)], set() while q: word, step = q.pop(0) if word not in visited: visited.add(word) if word == endWord: return step for i in range(len(word)): for j in 'abcdefghijklmnopqrstuvwxyz': tmp = word[:i] + j + word[i+1:] if tmp in wordDict and tmp not in visited: # difference q.append((tmp, step + 1)) return 0
上面的写法和传统的bfs
也一点不同,不同点我在代码中已经标出出来了。我们这里对于每个tmp
也判断了一次是不是在visited
中,而真正的bfs
算法不用这么做,这么做是为了效率。例如:
beginWord = "hit",
endWord = "cog",
wordList = ["hot","dot","dog","lot","log","cog"]
两组解:[
["hit","hot","dot","dog","cog"],
["hit","hot","lot","log","cog"]
]
如果采用了上述写法的话,只会产生一组解,问题在"hot"
处就把另一组解屏蔽了。
但是这显然不是最好的做法,我们这里并不是要将单词中的每个字母用26个字母替换一遍,而是只要用特殊字符替换即可。例如hot
_ot h_t ho_
只会出现上述三种情况。所以我们对给定的输入hit
也做相同的替换
_it h_t hi_
我们看hit
的替换是不是出现在dict
中,如果是的话,说明hit->hot
是可行的,我们要判断这个路径之前有没有访问过,如果没访问过的话,我们将h_t
加入q
,同时我们要更新我们的step
。
class Solution: def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int: wordDict = collections.defaultdict(list) for word in wordList: for i in range(len(word)): tmp = word[:i] + "_" + word[i+1:] wordDict[tmp].append(word) q, visited = [(beginWord, 1)], set() while q: word, step = q.pop(0) if word not in visited: visited.add(word) if word == endWord: return step for i in range(len(word)): tmp = word[:i] + "_" + word[i+1:] for neigh in wordDict[tmp]: q.append((neigh, step + 1)) return 0
这里我们也可以使用双向广度优先搜索。关于双向广度优先搜索其实非常简单,我们传统的广度优先搜索是从start->end
,而双向的是start-><-end
。我们首先建立一个stack
用来存储每次访问的元素,然后先从start
开始
start: hit
end: cog
wordDict: hot dot dog lot log
stack:
我们首先将start
中的每个元素从wordDict
中移除。然后将start
中的每个元素的每个位置替换为26个字母,然后判断替换后的单词tmp
是不是在wordDict
中,如果不在就继续替换,如果在的话,我们就判断tmp
在不在end
中,如果在的话,我们返回step+1
即可,如果不在,我们将tmp
加入到stack
中。
start: hit
end: cog
wordDict: hot dot dog lot log
stack: hot
然后我们判断stack.size() < end.size()
,如果是的话,我们将start
替换为stack
,不是的话,我们将start
替换为end
并且将end
替换为stack
。同时我们要将step+1
start: cog
end: hot
wordDict: hot dot dog lot log
stack: hot
现在我们就相当于cog->hot
寻找最短路径,也就是开始从后向前查找。重复上述操作直到start
和end
都是空,此时我们应该返回0
(因为找不到路径)。关键点在于start
和end
的交换,也就是判断stack.size() < end.size()
。这一步主要含义就是判断start
和end
谁的下一步可选范围更小,我们希望从可选范围小的那一方开始搜索。
class Solution: def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int: wordDict, step = set(wordList), 1 if endWord not in wordDict: return 0 s1, s2 = set([beginWord]), set([endWord]) while s1: stack = set([]) wordDict -= s1 for s in s1: for i in range(len(beginWord)): for j in string.ascii_lowercase: tmp = s[:i] + j + s[i+1:] if tmp not in wordDict: continue if tmp in s2: return step + 1 stack.add(tmp) if len(stack) < len(s2): s1 = stack else: s1, s2 = s2, stack step += 1 return 0
reference:
https://leetcode.com/problems/word-ladder/discuss/40723/Simple-to-understand-Python-solution-using-list-preprocessing-and-BFS-beats-95
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。