Skip to content

Commit 3166339

Browse files
committed
refactor(trie): fix none check
1 parent cb0b9c4 commit 3166339

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

datastructures/trees/trie/trie.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def insert(self, word: str, index: Optional[int] = None) -> None:
4343

4444
for char in word:
4545
curr = curr.children[char]
46-
if index:
46+
if index is not None:
4747
curr.index = min(curr.index or float("inf"), index)
4848

4949
curr.is_end = True

datastructures/trees/trie/trie_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,4 +59,4 @@ def search_prefix(self, prefix: str) -> int:
5959
# Traverse to the next node
6060
current = current.children[char]
6161

62-
return current.index
62+
return current.index if current.index is not None else -1

0 commit comments

Comments
 (0)