Skip to the content.

leetcode [127] 单词接龙


Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub


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

参考题解

原本的思路是建立无向图,两个单词有一个字母不同就设为连接,然后广度优先遍历,但是问题是,每次判断两个单词是否相邻复杂度过高,因此改为将每个单词加入到一个中间词的集合中,如a*b中可以包括[acb,adb]等。这样就不用平方复杂度建立无向图了。后面依旧采用广度优先搜索方法。

class Solution:
    def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int:
        if endWord not in wordList:
            return 0
        visited = defaultdict(bool)
        inter_word = defaultdict(list)
        for wd in wordList:
            for i in range(len(wd)):
                inter_word[wd[:i] + '*' + wd[i + 1:]].append(wd)

        level, leng = [beginWord], 0

        while level:
            newlevel = []
            for lev in level:
                if lev == endWord:
                    return leng + 1
                for i in range(len(lev)):
                    for iw in inter_word[lev[:i] + '*' + lev[i + 1:]]:
                        if not visited[iw]:
                            visited[iw] = True
                            newlevel.append(iw)

            level = newlevel
            leng += 1
        return 0