题目

127. 单词接龙

字典 wordList 中从单词 beginWord 到 endWord 的 转换序列 是一个按下述规格形成的序列 beginWord -> s1 -> s2 -> … -> sk：

每一对相邻的单词只差一个字母。
对于 1 <= i <= k 时，每个 si 都在 wordList 中。注意， beginWord 不需要在 wordList 中。
sk == endWord
给你两个单词 beginWord 和 endWord 和一个字典 wordList ，返回 从 beginWord 到 endWord 的 最短转换序列 中的 单词数目 。如果不存在这样的转换序列，返回 0 。

示例 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” 不在字典中，所以无法进行转换。

题解

class Solution(object):
    def ladderLength(self, beginWord, endWord, wordList):
        """
        :type beginWord: str
        :type endWord: str
        :type wordList: List[str]
        :rtype: int
        """
        if endWord not in wordList:
            return 0

        queue = collections.deque()
        queue.append([beginWord, 1])
        visited = set([beginWord])
        wordList = set(wordList)

        while queue:
            current, step = queue.popleft()
            if current == endWord:
                return step

            for i in range(len(beginWord)):
                for ch in 'abcdefghijklmnopqrstuvwxyz':
                    nextWord = current[:i] + ch + current[i + 1:]
                    if nextWord in wordList and nextWord not in visited:
                        visited.add(nextWord)
                        queue.append((nextWord, step + 1))
        return 0

    def diff(self, s1, s2):
        res = 0
        for i in range(len(s1)):
            if s1[i] != s2[i]:
                res += 1
        return res