-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy path127. Word Ladder
58 lines (47 loc) · 1.99 KB
/
127. Word Ladder
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
class Solution:
def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int:
if endWord not in wordList or not endWord or not beginWord or not wordList:
return 0
L = len(beginWord)
all_combo_dict = defaultdict(list)
for word in wordList:
for i in range(L):
all_combo_dict[word[:i] + "*" + word[i+1:]].append(word)
#print(all_combo_dict)
q = [(beginWord,1)]
visited = {beginWord:True}
while q:
current_word, level = q.pop(0)
for i in range(L):
intermediate_word = current_word[:i] + "*" + current_word[i+1:]
for word in all_combo_dict[intermediate_word]:
if word == endWord:
return level+1
if word not in visited:
visited[word] = True
q.append((word, level +1))
all_combo_dict[intermediate_word] = []
return 0
class Solution:
def ladderLength(self, beginWord: str, endWord: str, wordList: List[str]) -> int:
if endWord not in wordList or not endWord or not beginWord or not wordList:
return 0
d = defaultdict(list)
L = len(beginWord)
for word in wordList:
for i in range(L):
d[word[:i]+'*'+word[i+1:]].append(word)
q = [(beginWord,1)]
visited = set()
visited.add(beginWord)
while q:
curr,level = q.pop(0)
for i in range(L):
immediateWord = curr[:i] + '*' + curr[i+1:]
for word in d[immediateWord]:
if word == endWord:
return level +1
if word not in visited:
visited.add(word)
q.append((word,level+1))
return 0