leetcode127. 单词接龙

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/sinat_36811967/article/details/88528356

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

这里采用换一个字符的方法，因为不可能再去找以前找过的所以要去重，最后把小set放前面找会检索的更快：

class Solution:
    def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int:
        wordSet, beginSet, endSet, step = set(wordList), set([beginWord]), set([endWord]), 1
        if endWord not in wordList:
            return 0
        while beginSet:
            tmpSet, step = set(), step+1
            for word in beginSet:  # 去掉重复的
                if word in wordSet:
                    wordSet.remove(word)
            for word in beginSet:
                for i in range(len(word)):
                    tmp = list(word)
                    for c in list(map(chr, range(ord('a'), ord('z') + 1))):
                        tmp[i] = c  # 将tmp某一位替换
                        str_tmp = ''.join(tmp)
                        if str_tmp in endSet:
                            return step
                        if str_tmp in wordSet:
                            tmpSet.add(str_tmp)
            if len(tmpSet) > len(endSet):  # 把小set放前面检索的少
                beginSet, endSet = endSet, tmpSet
            else:
                beginSet = tmpSet
        return 0