[Leetcode]WordLadder有人用python不超时吗

如题。之前用C++写的时候就容易超时。用python试了几下没有通过。还是超时。
https://github.com/Linzertorte/L ... aster/WordLadder.py

1500 ms. 慢。 过了test


class Solution:
    # @param start, a string
    # @param end, a string
    # @param dict, a set of string
    # @return an integer
    def ladderLength(self, start, end, dict):
        dict.add(end)
        wordLen = len(start)
        candidates = set()
        candidates.add((start, 1))
        while True:
            if len(candidates)==0:  break
            current = set()
            for cur in candidates:
                curWord = cur[0];curLen = cur[1]
                if curWord == end:return curLen
                for i in range(wordLen):
                    part1 = curWord[:i]; part2 = curWord[i+1:] # replace ith char
                    #print part1, part2
                    for j in 'abcdefghijklmnopqrstuvwxyz': # BFS
                        if curWord[i] !=j:
                            nextWord = part1 + j + part2
                            #print nextWord
                            if nextWord == end: return curLen+1
                            if nextWord in dict:
                                current.add((nextWord, curLen + 1))
                                dict.remove(nextWord)
            candidates = current
        return 0

yzl232 发表于 2014-7-18 15:48
1500 ms. 慢。 过了test

厉害哦。

word ladder II也用python过掉了

这题挺卡python的，几个语句顺序不对就超时了

我的AC版本，交流学习吧


class Solution:
    # @param start, a string
    # @param end, a string
    # @param dict, a set of string
    # @return an integer
    def ladderLength(self, start, end, dict):
        if start == end:
            return 1
   
        all_chars = map(chr, xrange(ord('a'), ord('z') + 1))
        visited = set()
        bfs = [start, None]
        
        # at which distance the following to-be-processed nodes are
        distance = 2
        
        while len(bfs) > 1:
            word = bfs.pop(0)
            if word is None:
                distance += 1
                bfs.append(None)
                continue
            
            for i in xrange(len(word)):
                for char in all_chars:
                    new_word = word[:i] + char + word[i + 1:]
                    if new_word == end:
                        return distance
                    if new_word in dict and new_word not in visited:
                        bfs.append(new_word)
                        visited.add(new_word)
   
        return 0

        while True:
            if len(candidates)==0:  break
可以改成
        while candidates: