欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

leeetoe:单词接龙(python)

程序员文章站 2022-05-04 17:47:30
...

1. 题目描述

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

2. 思路

主要方法是利用队列实现广度优先搜索。找到字典里和当前单词相差一个字母的单词入队,然后遍历这几个单词,直到找到endWord。

  1. 先构造一个字符串队列,并将beginWord加入队列。
  2. 对队列头字符串做单个字符替换每次替换后,判断是否和endWord匹配,如果匹配,返回res,没有匹配,看一下字典里有没有这个单词。
  3. 如果有,则将该字符串加入队列,同时将该字符串从字典里删除。
  4. 重复,直到和endWord匹配。如果最后队列为空还未匹配到,则返回0.
如果直接在wordList中查找的话,会超时。set中找一个单词的时间复杂度是O(log(n)),比list快。
class Solution:
    def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int:
        if endWord not in wordList:
            return 0
        queue = [beginWord]
        res = 1
        wordList = set(wordList)
        while queue:
            for a in range(len(queue)):
                curstr = queue.pop(0)
                curstrlen = len(curstr)
                for i in range(curstrlen):
                    for x in range(ord('a'), ord('z') + 1):   
                    # 将二十六个英文字母分别和当前字符串的某一个字母来替换
                        temp = curstr[:i] + chr(x) + curstr[i+1:]
                        if temp == endWord and temp in wordList:
                            return res + 1
                        elif temp != endWord and temp in wordList:
                            queue.append(temp)
                            wordList.remove(temp)
            res += 1
        return 0
相关标签: 单词接龙