leetcode127. 单词接龙

题目：

给定两个单词（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” 不在字典中，所以无法进行转换。

思路：BFS

跟上一题思路一样，不用深搜找路径了，用一个变量记录层数即可。

代码：

class Solution(object):
    def ladderLength(self, beginWord, endWord, wordList):
        """
        :type beginWord: str
        :type endWord: str
        :type wordList: List[str]
        :rtype: int
        """
        def getNextWords(curWord, wordSet):
            res = set()
            for i in range(len(curWord)):
                for j in string.lowercase:
                    s = curWord[0:i] + j + curWord[i + 1:len(curWord)]
                    if s == curWord or s not in wordSet:
                        continue
                    res.add(s)
            return res
        def bfs(beginWord, endWord, visited, wordSet):
            queue = []

            queue.append(beginWord)
            minLength = 1
            found = False
            while queue:
                l = len(queue)

                for i in queue:
                    visited.add(i)


                for i in range(l):
                    cur = queue.pop(0)

                    for j in iter(getNextWords(cur, wordSet)):
                        if j in visited:
                            continue
                        queue.append(j)
                        if j == endWord:
                            found = True
                minLength += 1
                if found:
                    return minLength
            return 0

        wordSet = set(wordList)
        visited = set()

        return bfs(beginWord, endWord, visited, wordSet)