美文网首页
126. Word Ladder II

126. Word Ladder II

作者: April63 | 来源:发表于2018-06-14 20:59 被阅读0次

这一题一个AC了的做法,明天早上研究下:

class Solution:
      def findLadders(self, beginWord, endWord, wordList):
        """
        :type beginWord: str
        :type endWord: str
        :type wordList: List[str]
        :rtype: List[List[str]]
        """
        
        wdset=set(wordList)
        if endWord not in wdset:
            return []
        lts="abcdefghijklmnopqrstuvwxyz"
        dist=float("inf")
        q=[beginWord] 
        seen={beginWord:0}
        graph={beginWord:set()}
        while q:
            cur=q.pop(0)
            d=seen[cur]
            if d>=dist:
                break
            for i in range(len(cur)):
                for lt in lts:
                    if lt!=cur[i]:
                        new=cur[:i]+lt+cur[i+1:]
                        if new in wdset and (new not in seen or d+1==seen[new]):
                            if cur in graph:
                                graph[cur].add(new)
                            else:
                                graph[cur]={new}
                            if new==endWord:
                                dist=d+1
                            if new not in seen:
                                seen[new]=d+1
                                q.append(new)
        res=[]
        def dfs(path,cur):
            if cur==endWord:
                res.append(path[:])
            else:
                if cur in graph:
                    for new in graph[cur]:
                        dfs(path+[new],new)
        dfs([beginWord],beginWord)
        return res

相关文章

网友评论

      本文标题:126. Word Ladder II

      本文链接:https://www.haomeiwen.com/subject/qyjueftx.html