leetcode - 126. Word Ladder II

Description

A transformation sequence from word beginWord to word endWord using a dictionary wordList is a sequence of words beginWord -> s1 -> s2 -> ... -> sk such that:

Every adjacent pair of words differs by a single letter.

Every si for 1 <= i <= k is in wordList. Note that beginWord does not need to be in wordList.

sk == endWord

Given two words, beginWord and endWord, and a dictionary wordList, return all the shortest transformation sequences from beginWord to endWord, or an empty list if no such sequence exists. Each sequence should be returned as a list of the words [beginWord, s1, s2, ..., sk].

Example 1:

复制代码
Input: beginWord = "hit", endWord = "cog", wordList = ["hot","dot","dog","lot","log","cog"]
Output: [["hit","hot","dot","dog","cog"],["hit","hot","lot","log","cog"]]
Explanation: There are 2 shortest transformation sequences:
"hit" -> "hot" -> "dot" -> "dog" -> "cog"
"hit" -> "hot" -> "lot" -> "log" -> "cog"

Example 2:

复制代码
Input: beginWord = "hit", endWord = "cog", wordList = ["hot","dot","dog","lot","log"]
Output: []
Explanation: The endWord "cog" is not in wordList, therefore there is no valid transformation sequence.

Constraints:

复制代码
1 <= beginWord.length <= 5
endWord.length == beginWord.length
1 <= wordList.length <= 500
wordList[i].length == beginWord.length
beginWord, endWord, and wordList[i] consist of lowercase English letters.
beginWord != endWord
All the words in wordList are unique.
The sum of all shortest transformation sequences does not exceed 10^5.

Solution

TLE: bfs + backtracing. Similar to 127. Word Ladder, this time we will need to print out the paths. So I will call 127's function first, to get the shortest path length, and then use dfs/back tracking to find the paths that has this length.

Code

python3 复制代码
class Solution:
    def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int:
        word_list = set(wordList)
        if endWord not in word_list:
            return 0
        queue = collections.deque([(endWord, 1)])
        visited = set()
        while queue:
            cur_word, step = queue.popleft()
            if cur_word in visited:
                continue
            visited.add(cur_word)
            if cur_word == beginWord:
                return step
            for i in range(len(cur_word)):
                for dz in range(26):
                    new_char = chr(ord('a') + dz)
                    if new_char == cur_word[i]:
                        continue
                    new_word = f'{cur_word[:i]}{new_char}{cur_word[i+1:]}'
                    if new_word in word_list or new_word == beginWord:
                        queue.append((new_word, step + 1))
        return 0
    def findLadders(self, beginWord: str, endWord: str, wordList: List[str]) -> List[List[str]]:
        ladder_length = self.ladderLength(beginWord, endWord, wordList)
        if ladder_length == 0:
            return []
        res = []
        word_list = set(wordList)
        candidates = {}
        for each_word in word_list | set([beginWord]):
            candidates[each_word] = []
            for i in range(len(each_word)):
                for dz in range(26):
                    new_char = chr(ord('a') + dz)
                    if new_char == each_word[i]:
                        continue
                    new_word = f'{each_word[:i]}{new_char}{each_word[i+1:]}'
                    if new_word in word_list:
                        candidates[each_word].append(new_word)
        def dfs(cur_word: str, cur_res: list) -> None:
            if cur_word == endWord and len(cur_res) == ladder_length:
                res.append(cur_res)
            elif len(cur_res) > ladder_length:
                return
            else:
                for candidate_word in candidates[cur_word]:
                    dfs(candidate_word, cur_res + [candidate_word])
        dfs(beginWord, [beginWord])
        return res
相关推荐
好望角雾眠1 小时前
第四阶段C#通讯开发-1:通讯基础理论,串口,通讯模式,单位转换,代码示例
开发语言·笔记·c#·串口·通讯
·云扬·3 小时前
【Leetcode hot 100】101.对称二叉树
算法·leetcode·职场和发展
睡不醒的kun8 小时前
leetcode算法刷题的第三十二天
数据结构·c++·算法·leetcode·职场和发展·贪心算法·动态规划
薄荷撞~可乐8 小时前
C#Task(Api)应用
开发语言·c#
共享家952713 小时前
经典动态规划题解
算法·leetcode·动态规划
almighty2714 小时前
C#WPF控制USB摄像头参数:曝光、白平衡等高级设置完全指南
开发语言·c#·wpf·usb相机·参数设置
后青春期的诗go14 小时前
金蝶云星空插件开发记录(一)
c#·钉钉·金蝶云星空·插件开发
1白天的黑夜114 小时前
栈-844.比较含退格的字符串-力扣(LeetCode)
c++·leetcode·
Swift社区16 小时前
LeetCode 378 - 有序矩阵中第 K 小的元素
算法·leetcode·矩阵